Home
last modified time | relevance | path

Searched defs:input_flat (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Das_string_op.cc159 const auto& input_flat = input_tensor->flat<bool>(); in Compute() local
165 const auto& input_flat = input_tensor->flat<Variant>(); in Compute() local
171 const auto& input_flat = input_tensor->flat<complex64>(); in Compute() local
178 const auto& input_flat = input_tensor->flat<complex128>(); in Compute() local
Dregex_full_match_op.cc38 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
102 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
Dstring_to_number_op.cc43 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
Dunicode_script_op.cc29 const auto& input_flat = input_tensor->flat<int32>(); in Compute() local
Dstring_to_hash_bucket_fast_op.h39 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
Dstring_to_hash_bucket_op.cc35 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
Dstring_to_hash_bucket_op.h46 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
Ddebug_ops.h245 const T* input_flat = input.template flat<T>().data(); in Compute() local
301 const T* input_flat = input.template flat<T>().data(); in Compute() local
Dquantized_batch_norm_op.cc39 auto input_flat = input.flat<T1>(); in ReferenceBatchNorm() local
102 auto input_flat = input.flat<T1>(); in FixedPointBatchNorm() local
Dsegment_reduction_ops_impl.h100 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
Dreduce_join_op.cc125 const auto input_flat = input.flat<tstring>(); in Compute() local
Droll_op.cc103 auto input_flat = input.flat<T>().data(); in Compute() local
Dfake_quant_ops_test.cc380 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
410 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
441 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
472 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
Dquantization_utils.h819 const auto& input_flat = input.flat<T1>(); in QuantizedAddUsingEigen() local
865 const auto& input_flat = input.flat<T1>(); in QuantizedAdd() local
Dset_kernels.cc212 auto input_flat = input_tensor.flat<T>(); in PopulateFromDenseGroup() local
/external/tensorflow/tensorflow/lite/python/testdata/
Ddouble_op.cc36 auto input_flat = input_tensor.flat<int32>(); in Compute() local
/external/tensorflow/tensorflow/core/kernels/image/
Dadjust_contrast_op.cc238 TTypes<float, 1>::ConstTensor input_flat(&input(0, 0, 0), input.size()); in ReduceMeanAcrossImage() local