| /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),
56 errors::InvalidArgument(kErrorMessage, input_flat(i).c_str()));
|
| unicode_script_op.cc | 29 const auto& input_flat = input_tensor->flat<int32>(); variable
38 for (int i = 0; i < input_flat.size(); i++) {
39 UScriptCode script_code = uscript_getScript(input_flat(i), status);
|
| regex_full_match_op.cc | 34 const auto& input_flat = input_tensor->flat<string>(); variable
51 for (size_t i = 0; i < input_flat.size(); ++i) {
52 output_flat(i) = RE2::FullMatch(input_flat(i), match);
74 const auto& input_flat = input_tensor->flat<string>(); variable
80 for (size_t i = 0; i < input_flat.size(); ++i) {
81 output_flat(i) = RE2::FullMatch(input_flat(i), *re_);
|
| 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 | 123 const auto& input_flat = input_tensor->flat<T>(); \
124 for (int i = 0; i < input_flat.size(); ++i) { \
125 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \
137 const auto& input_flat = input_tensor->flat<bool>(); variable
138 for (int i = 0; i < input_flat.size(); ++i) {
139 output_flat(i) = (input_flat(i)) ? "true" : "false";
143 const auto& input_flat = input_tensor->flat<complex64>(); variable
144 for (int i = 0; i < input_flat.size(); ++i) {
146 format_.c_str(), input_flat(i).real(), input_flat(i).imag())
150 const auto& input_flat = input_tensor->flat<complex128>(); variable
[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 | 292 auto input_flat = input.flat<T>().data(); variable
299 input_flat, output_flat, threshold, dim_range, isd);
302 DoRoll<T>(context, num_elements, num_dims, dim_size, input_flat,
|
| debug_ops.h | 246 const T* input_flat = input.template flat<T>().data(); variable
249 if (Eigen::numext::isnan(static_cast<double>(input_flat[i]))) {
302 const T* input_flat = input.template flat<T>().data(); variable
309 const double x = static_cast<double>(input_flat[i]);
349 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 | 91 auto input_flat = input.flat_outer_dims<T>(); variable
92 const int64 num_col = input_flat.dimension(1);
147 const T* in_slice_ptr = &input_flat(start, 0);
658 auto input_flat = input.flat_outer_dims<T>(); variable
659 const int64 num_col = input_flat.dimension(1);
735 Reduce(input_flat, indices_vec, start, end - start, out);
1032 auto input_flat = input.flat_outer_dims<T>(); variable
[all...] |
| set_kernels.cc | 212 auto input_flat = input_tensor.flat<T>(); local
218 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), "\""));
|
