Home
last modified time | relevance | path

Searched refs:input_flat (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Das_string_op.cc145 const auto& input_flat = input_tensor->flat<T>(); \ in Compute()
146 for (int i = 0; i < input_flat.size(); ++i) { \ in Compute()
147 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \ in Compute()
159 const auto& input_flat = input_tensor->flat<bool>(); in Compute() local
160 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
161 output_flat(i) = (input_flat(i)) ? "true" : "false"; in Compute()
165 const auto& input_flat = input_tensor->flat<Variant>(); in Compute() local
166 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
167 output_flat(i) = input_flat(i).DebugString(); in Compute()
171 const auto& input_flat = input_tensor->flat<complex64>(); in Compute() local
[all …]
Dregex_full_match_op.cc38 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
55 for (size_t i = 0; i < input_flat.size(); ++i) { in Compute()
56 output_flat(i) = RE2::FullMatch(input_flat(i), *regex); in Compute()
102 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
108 for (size_t i = 0; i < input_flat.size(); ++i) { in Compute()
109 output_flat(i) = RE2::FullMatch(input_flat(i), *re_); in Compute()
Dstring_to_number_op.cc43 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
51 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
54 strings::SafeStringToNumeric<OutputType>(input_flat(i), in Compute()
56 errors::InvalidArgument(kErrorMessage, input_flat(i).c_str())); in Compute()
Dstring_to_hash_bucket_op.cc35 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
43 typedef decltype(input_flat.size()) Index; in Compute()
44 for (Index i = 0; i < input_flat.size(); ++i) { in Compute()
45 const uint64 input_hash = Hash64(input_flat(i)); in Compute()
Dstring_to_hash_bucket_fast_op.h39 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
47 typedef decltype(input_flat.size()) Index; in Compute()
48 for (Index i = 0; i < input_flat.size(); ++i) { in Compute()
49 const uint64 input_hash = hash(input_flat(i)); in Compute()
Dstring_to_hash_bucket_op.h46 const auto& input_flat = input_tensor->flat<tstring>(); in Compute() local
54 typedef decltype(input_flat.size()) Index; in Compute()
55 for (Index i = 0; i < input_flat.size(); ++i) { in Compute()
56 const uint64 input_hash = hash(key_, input_flat(i)); in Compute()
Dsegment_reduction_ops_impl.h100 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
101 const int64 num_col = input_flat.dimension(1); in Compute()
156 const T* in_slice_ptr = &input_flat(start, 0); in Compute()
476 auto input_flat = input.flat_outer_dims<T>();
477 const int64 num_col = input_flat.dimension(1);
558 const int bad_offset = Reduce<T, Index>(input_flat, indices_vec, start,
564 " out of range [0, ", input_flat.dimension(0), ")"));
593 const typename TTypes<Tin>::ConstMatrix& input_flat, Tindex index) {
594 return input_flat.template chip<0>(index);
599 const typename TTypes<Tin>::ConstMatrix& input_flat, Tindex index) {
[all …]
Dunicode_script_op.cc29 const auto& input_flat = input_tensor->flat<int32>(); in Compute() local
38 for (int i = 0; i < input_flat.size(); i++) { in Compute()
39 UScriptCode script_code = uscript_getScript(input_flat(i), status); in Compute()
Dquantized_batch_norm_op.cc39 auto input_flat = input.flat<T1>(); in ReferenceBatchNorm() local
47 const int row_count = input_flat.size() / depth; in ReferenceBatchNorm()
57 QuantizedToFloat(input_flat(input_index), input_min, input_max); in ReferenceBatchNorm()
102 auto input_flat = input.flat<T1>(); in FixedPointBatchNorm() local
110 const int row_count = input_flat.size() / depth; in FixedPointBatchNorm()
150 RequantizeInNewRange<T1, T2>(input_flat(input_index), input_min, in FixedPointBatchNorm()
Dfake_quant_ops_test.cc380 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
382 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2), in TEST_F()
383 input_flat(3), input_flat(4), 0.0f}); in TEST_F()
410 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
412 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2), in TEST_F()
413 input_flat(3), input_flat(4), 0.0f}); in TEST_F()
441 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
443 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2), in TEST_F()
444 input_flat(3), input_flat(4), 0.0f}); in TEST_F()
472 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
[all …]
Dreduce_join_op.cc125 const auto input_flat = input.flat<tstring>(); in Compute() local
173 input_flat(output_full_index + reduction_full_index); in Compute()
Ddebug_ops.h245 const T* input_flat = input.template flat<T>().data(); in Compute() local
248 if (Eigen::numext::isnan(static_cast<double>(input_flat[i]))) { in Compute()
301 const T* input_flat = input.template flat<T>().data(); in Compute() local
308 const double x = static_cast<double>(input_flat[i]); in Compute()
348 const double x = static_cast<double>(input_flat[i]); in Compute()
Droll_op.cc103 auto input_flat = input.flat<T>().data(); in Compute() local
107 input_flat, output_flat, threshold, dim_range, in Compute()
Dquantization_utils.h819 const auto& input_flat = input.flat<T1>(); in QuantizedAddUsingEigen() local
845 auto input_float = DEQUANTIZE_WITH_EIGEN(input_flat, input_q2f); in QuantizedAddUsingEigen()
865 const auto& input_flat = input.flat<T1>(); in QuantizedAdd() local
893 const T1 input_value = input_flat(index); in QuantizedAdd()
Dset_kernels.cc212 auto input_flat = input_tensor.flat<T>(); in PopulateFromDenseGroup() local
218 result->insert(input_flat(i)); in PopulateFromDenseGroup()
/external/tensorflow/tensorflow/lite/python/testdata/
Ddouble_op.cc36 auto input_flat = input_tensor.flat<int32>(); in Compute() local
45 const int N = input_flat.size(); in Compute()
47 output_flat(i) = 2 * input_flat(i); in Compute()
/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
250 input_flat.slice(Index(i * plane_size), Index(plane_size)); in ReduceMeanAcrossImage()
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py1497 input_flat = nest.flatten(input_ct, expand_composites=True)
1499 for (input_component, output_component) in zip(input_flat, output_flat):
1543 input_flat = nest.flatten(input_ct, expand_composites=True)
1545 for (input_component, output_component) in zip(input_flat, output_flat):