| /external/tensorflow/tensorflow/core/kernels/ |
| D | as_string_op.cc | 120 const auto& input_flat = input_tensor->flat<T>(); \ in Compute()
121 for (int i = 0; i < input_flat.size(); ++i) { \ in Compute()
122 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \ in Compute()
133 const auto& input_flat = input_tensor->flat<bool>(); in Compute() local
134 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
135 output_flat(i) = (input_flat(i)) ? "true" : "false"; in Compute()
139 const auto& input_flat = input_tensor->flat<complex64>(); in Compute() local
140 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
142 format_.c_str(), input_flat(i).real(), input_flat(i).imag()); in Compute()
|
| D | string_to_hash_bucket_op.h | 39 const auto& input_flat = input_tensor->flat<string>(); 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()
81 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
89 typedef decltype(input_flat.size()) Index; in Compute()
90 for (Index i = 0; i < input_flat.size(); ++i) { in Compute()
91 const uint64 input_hash = hash(key_, input_flat(i)); in Compute()
|
| D | string_to_number_op.cc | 43 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
51 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
54 strings::SafeStringToNumeric<OutputType>(input_flat(i).c_str(), in Compute()
56 errors::InvalidArgument(kErrorMessage, input_flat(i).c_str())); in Compute()
|
| D | string_to_hash_bucket_op.cc | 36 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
44 typedef decltype(input_flat.size()) Index; in Compute()
45 for (Index i = 0; i < input_flat.size(); ++i) { in Compute()
46 const uint64 input_hash = Hash64(input_flat(i)); in Compute()
|
| D | quantized_batch_norm_op.cc | 39 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()
|
| D | segment_reduction_ops.cc | 89 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
90 const int64 num_col = input_flat.dimension(1); in Compute()
145 const T* in_slice_ptr = &input_flat(start, 0); in Compute()
654 auto input_flat = input.flat_outer_dims<T>(); in Compute() local
655 const int64 num_col = input_flat.dimension(1); in Compute()
726 Reduce(input_flat, indices_vec, start, end - start, out); in Compute()
731 " out of range [0, ", input_flat.dimension(0), ")")); in Compute()
753 int64 Reduce(const typename TTypes<T>::ConstMatrix& input_flat, in Reduce() argument
759 if (!FastBoundsCheck(index##n, input_flat.dimension(0))) return (i); in Reduce()
761 #define L(n) input_flat.template chip<0>(index##n) in Reduce()
[all …]
|
| D | fake_quant_ops_test.cc | 379 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
381 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2), in TEST_F()
382 input_flat(3), input_flat(4), 0.0f}); in TEST_F()
409 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
411 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2), in TEST_F()
412 input_flat(3), input_flat(4), 0.0f}); in TEST_F()
440 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
442 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2), in TEST_F()
443 input_flat(3), input_flat(4), 0.0f}); in TEST_F()
471 auto input_flat = GetInput(0).flat<float>(); in TEST_F() local
[all …]
|
| D | debug_ops.h | 243 const T* input_flat = input.template flat<T>().data(); in Compute() local
246 if (Eigen::numext::isnan(static_cast<double>(input_flat[i]))) { in Compute()
299 const T* input_flat = input.template flat<T>().data(); in Compute() local
306 const double x = static_cast<double>(input_flat[i]); in Compute()
346 const double x = static_cast<double>(input_flat[i]); in Compute()
|
| D | roll_op.cc | 288 auto input_flat = input.flat<T>().data(); in Compute() local
295 input_flat, output_flat, threshold, dim_range, isd); in Compute()
298 DoRoll<T>(context, num_elements, num_dims, dim_size, input_flat, in Compute()
|
| D | reduce_join_op.cc | 125 const auto input_flat = input.flat<string>(); in Compute() local
173 input_flat(output_full_index + reduction_full_index); in Compute()
|
| D | adjust_contrast_op.cc | 238 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()
|
| D | quantization_utils.h | 810 const auto& input_flat = input.flat<T1>(); in QuantizedAddUsingEigen() local
836 auto input_float = DEQUANTIZE_WITH_EIGEN(input_flat, input_q2f); in QuantizedAddUsingEigen()
856 const auto& input_flat = input.flat<T1>(); in QuantizedAdd() local
884 const T1 input_value = input_flat(index); in QuantizedAdd()
|
| D | set_kernels.cc | 217 auto input_flat = input_tensor.flat<T>(); in PopulateFromDenseGroup() local
223 result->insert(input_flat(i)); in PopulateFromDenseGroup()
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| D | decode_libsvm_op.cc | 39 const auto& input_flat = input_tensor->flat<string>(); in Compute() local
48 for (int i = 0; i < input_flat.size(); ++i) { in Compute()
49 StringPiece line(input_flat(i)); in Compute()
55 "]: \"", input_flat(i), "\"")); in Compute()
|