Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Das_string_op.cc137 const auto& input_flat = input_tensor->flat<bool>(); in Compute() local
143 const auto& input_flat = input_tensor->flat<complex64>(); in Compute() local
150 const auto& input_flat = input_tensor->flat<complex128>(); in Compute() local
Dregex_full_match_op.cc34 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
74 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
Dstring_to_hash_bucket_op.h39 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
81 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
Dstring_to_number_op.cc43 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
Dunicode_script_op.cc29 const auto& input_flat = input_tensor->flat<int32>(); in Compute() local
Dstring_to_hash_bucket_op.cc36 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
Ddebug_ops.h246 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
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.cc91 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
658 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
762 int64 Reduce(const typename TTypes<T>::ConstMatrix& input_flat, in Reduce()
1032 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
Dreduce_join_op.cc125 const auto input_flat = input.flat<string>(); in Compute() local
Droll_op.cc292 auto input_flat = input.flat<T>().data(); in Compute() local
Dfake_quant_ops_test.cc379 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
409 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
440 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
471 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
Dadjust_contrast_op.cc238 TTypes<float, 1>::ConstTensor input_flat(&input(0, 0, 0), input.size()); in ReduceMeanAcrossImage() local
Dquantization_utils.h810 const auto& input_flat = input.flat<T1>(); in QuantizedAddUsingEigen() local
856 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/contrib/libsvm/kernels/
Ddecode_libsvm_op.cc39 const auto& input_flat = input_tensor->flat<string>(); in Compute() local