| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | as_string_op.cc | 175 const auto& input_flat = input_tensor->flat<bool>(); in Compute() local
181 const auto& input_flat = input_tensor->flat<Variant>(); in Compute() local
187 const auto& input_flat = input_tensor->flat<Eigen::half>(); in Compute() local
194 const auto& input_flat = input_tensor->flat<bfloat16>(); in Compute() local
201 const auto& input_flat = input_tensor->flat<complex64>(); in Compute() local
208 const auto& input_flat = input_tensor->flat<complex128>(); in Compute() local
|
| H A D | regex_full_match_op.cc | 38 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
102 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
|
| H A D | debug_ops.h | 246 const T* input_flat = input.template flat<T>().data(); in Compute() local
302 const T* input_flat = input.template flat<T>().data(); in Compute() local
|
| H A D | quantized_batch_norm_op.cc | 39 auto input_flat = input.flat<T1>(); in ReferenceBatchNorm() local
102 auto input_flat = input.flat<T1>(); in FixedPointBatchNorm() local
|
| H A D | fake_quant_ops_test.cc | 450 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
480 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
511 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
542 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
|
| H A D | string_to_number_op.cc | 43 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
|
| H A D | unicode_script_op.cc | 29 const auto& input_flat = input_tensor->flat<int32>(); in Compute() local
|
| H A D | string_to_hash_bucket_fast_op.h | 39 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
|
| H A D | string_to_hash_bucket_op.h | 46 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
|
| H A D | string_to_hash_bucket_op.cc | 35 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
|
| H A D | tensor_to_hash_bucket_op.cc | 47 const auto& input_flat = input_tensor->flat<T>(); in Compute() local
|
| H A D | segment_reduction_ops_impl.h | 98 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
|
| H A D | reduce_join_op.cc | 126 const auto input_flat = input.flat<tstring>(); in Compute() local
|
| H A D | roll_op.cc | 103 auto input_flat = input.flat<T>().data(); in Compute() local
|
| H A D | quantization_utils.h | 817 const auto& input_flat = input.flat<T1>(); in QuantizedAddUsingEigen() local
863 const auto& input_flat = input.flat<T1>(); in QuantizedAdd() local
|
| H A D | set_kernels.cc | 224 auto input_flat = input_tensor.flat<T>(); in PopulateFromDenseGroup() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/testdata/ |
| H A D | double_op.cc | 38 auto input_flat = input_tensor.flat<T>(); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/calibrator/ |
| H A D | custom_aggregator_op.cc | 42 auto input_flat = input_tensor.flat<float>(); in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Unique.cu | 132 Tensor input_flat = self.moveaxis(dim, 0).contiguous().view({num_inp, -1}); in unique_dim_cuda_template() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | adjust_contrast_op.cc | 247 TTypes<float, 1>::ConstTensor input_flat(&input(0, 0, 0), input.size()); in ReduceMeanAcrossImage() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unique.cpp | 385 Tensor input_flat = self.moveaxis(dim, 0); in _unique_dim_cpu_template() local
|