Searched refs:ArrayRefTensor (Results 1 – 3 of 3) sorted by relevance
186 class ArrayRefTensor {188 ArrayRefTensor() = default;190 explicit ArrayRefTensor( in ArrayRefTensor() function273 sizeof(ArrayRefTensor<int>) ==275 (alignof(ArrayRefTensor<int>) > 4 ? sizeof(int32_t) : 0),291 inline ArrayRefTensor<T> reinterpret_tensor_wrapper( in reinterpret_tensor_wrapper()292 const ArrayRefTensor<T>& self, in reinterpret_tensor_wrapper()300 return ArrayRefTensor<T>( in reinterpret_tensor_wrapper()316 inline T* get_data_ptr_wrapper(ArrayRefTensor<T>& tensor) { in get_data_ptr_wrapper()331 inline const ArrayRefTensor<T>& unwrap_raii_handle_if_needed( in unwrap_raii_handle_if_needed()[all …]
47 struct ThreadLocalCachedOutputTensor<ArrayRefTensor<T>> {48 explicit ThreadLocalCachedOutputTensor(const ArrayRefTensor<T>& t) {52 void copy_data_from(const ArrayRefTensor<T>& t) {64 void realloc(const ArrayRefTensor<T>& t) {103 ArrayRefTensor<U> arrayref_tensor() const {122 ArrayRefTensor<U> arrayref_tensor() const {128 struct ThreadLocalCachedOutputArray<ArrayRefTensor<T>> {129 explicit ThreadLocalCachedOutputArray(const ArrayRefTensor<T>& t) {}136 ArrayRefTensor<T> arrayref_tensor() const {140 void copy_data_from(const ArrayRefTensor<T>& t) {[all …]
11 const ArrayRefTensor<T>& output, in convert_output_to_handle()18 const std::tuple<ArrayRefTensor<Ts>...>& outputs, in convert_outputs_to_handles_helper()25 const std::tuple<ArrayRefTensor<Ts>...>& outputs, in convert_outputs_to_handles()34 ArrayRefTensor<T>& input) { in convert_handle_to_arrayref_tensor()53 input = ArrayRefTensor<T>( in convert_handle_to_arrayref_tensor()64 std::tuple<ArrayRefTensor<Ts>...>& inputs, in convert_handles_to_inputs_helper()73 std::tuple<ArrayRefTensor<Ts>...>& inputs) { in convert_handles_to_inputs()79 void assert_numel(const ArrayRefTensor<T>& tensor, uint64_t numel) { in assert_numel()