| /external/tensorflow/tensorflow/core/kernels/ |
| string_to_number_op.cc | 43 const auto& input_flat = input_tensor->flat<string>(); variable
51 for (int i = 0; i < input_flat.size(); ++i) {
54 strings::SafeStringToNumeric<OutputType>(input_flat(i).c_str(),
56 errors::InvalidArgument(kErrorMessage, input_flat(i).c_str()));
|
| string_to_hash_bucket_op.cc | 36 const auto& input_flat = input_tensor->flat<string>(); variable
44 typedef decltype(input_flat.size()) Index;
45 for (Index i = 0; i < input_flat.size(); ++i) {
46 const uint64 input_hash = Hash64(input_flat(i));
|
| string_to_hash_bucket_op.h | 39 const auto& input_flat = input_tensor->flat<string>(); variable
47 typedef decltype(input_flat.size()) Index;
48 for (Index i = 0; i < input_flat.size(); ++i) {
49 const uint64 input_hash = hash(input_flat(i));
81 const auto& input_flat = input_tensor->flat<string>(); variable
89 typedef decltype(input_flat.size()) Index;
90 for (Index i = 0; i < input_flat.size(); ++i) {
91 const uint64 input_hash = hash(key_, input_flat(i));
|
| as_string_op.cc | 120 const auto& input_flat = input_tensor->flat<T>(); \
121 for (int i = 0; i < input_flat.size(); ++i) { \
122 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \
133 const auto& input_flat = input_tensor->flat<bool>(); variable
134 for (int i = 0; i < input_flat.size(); ++i) {
135 output_flat(i) = (input_flat(i)) ? "true" : "false";
139 const auto& input_flat = input_tensor->flat<complex64>(); variable
140 for (int i = 0; i < input_flat.size(); ++i) {
142 format_.c_str(), input_flat(i).real(), input_flat(i).imag())
[all...] |
| fake_quant_ops_test.cc | 379 auto input_flat = GetInput(0).flat<float>(); local
381 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2),
382 input_flat(3), input_flat(4), 0.0f});
409 auto input_flat = GetInput(0).flat<float>(); local
411 FillValues<float>(&expected, {0.0f, input_flat(1), input_flat(2),
412 input_flat(3), input_flat(4), 0.0f})
440 auto input_flat = GetInput(0).flat<float>(); local
471 auto input_flat = GetInput(0).flat<float>(); local
[all...] |
| reduce_join_op.cc | 125 const auto input_flat = input.flat<string>(); local
173 input_flat(output_full_index + reduction_full_index);
|
| roll_op.cc | 288 auto input_flat = input.flat<T>().data(); variable
295 input_flat, output_flat, threshold, dim_range, isd);
298 DoRoll<T>(context, num_elements, num_dims, dim_size, input_flat,
|
| debug_ops.h | 243 const T* input_flat = input.template flat<T>().data(); variable
246 if (Eigen::numext::isnan(static_cast<double>(input_flat[i]))) {
299 const T* input_flat = input.template flat<T>().data(); variable
306 const double x = static_cast<double>(input_flat[i]);
346 const double x = static_cast<double>(input_flat[i]);
|
| quantized_batch_norm_op.cc | 39 auto input_flat = input.flat<T1>(); local
47 const int row_count = input_flat.size() / depth;
57 QuantizedToFloat(input_flat(input_index), input_min, input_max);
102 auto input_flat = input.flat<T1>(); local
110 const int row_count = input_flat.size() / depth;
150 RequantizeInNewRange<T1, T2>(input_flat(input_index), input_min,
|
| segment_reduction_ops.cc | 89 auto input_flat = input.flat_outer_dims<T>(); variable
90 const int64 num_col = input_flat.dimension(1);
145 const T* in_slice_ptr = &input_flat(start, 0);
654 auto input_flat = input.flat_outer_dims<T>(); variable
655 const int64 num_col = input_flat.dimension(1);
726 Reduce(input_flat, indices_vec, start, end - start, out);
731 " out of range [0, ", input_flat.dimension(0), ")"));
1023 auto input_flat = input.flat_outer_dims<T>(); variable
[all...] |
| set_kernels.cc | 217 auto input_flat = input_tensor.flat<T>(); local
223 result->insert(input_flat(i));
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| decode_libsvm_op.cc | 39 const auto& input_flat = input_tensor->flat<string>(); variable
48 for (int i = 0; i < input_flat.size(); ++i) {
49 StringPiece line(input_flat(i));
55 "]: \"", input_flat(i), "\""));
|
