| /external/tensorflow/tensorflow/core/kernels/ |
| string_join_op.cc | 63 auto output_flat = output_tensor->flat<string>(); variable
70 output_flat(i) = str_util::Join(strings, separator_.c_str());
|
| string_to_number_op.cc | 49 auto output_flat = output_tensor->flat<OutputType>(); variable
55 &output_flat(i)),
|
| unicode_script_op.cc | 35 auto output_flat = output_tensor->flat<int32>(); variable
41 output_flat(i) = script_code;
43 output_flat(i) = -1;
|
| regex_full_match_op.cc | 50 auto output_flat = output_tensor->flat<bool>(); variable
52 output_flat(i) = RE2::FullMatch(input_flat(i), match);
79 auto output_flat = output_tensor->flat<bool>(); variable
81 output_flat(i) = RE2::FullMatch(input_flat(i), *re_);
|
| regex_replace_op.cc | 49 auto output_flat = output_tensor->flat<string>(); local
50 for (size_t i = 0; i < output_flat.size(); ++i) {
52 RE2::GlobalReplace(&output_flat(i), match, rewrite);
54 RE2::Replace(&output_flat(i), match, rewrite);
|
| string_to_hash_bucket_op.cc | 42 auto output_flat = output_tensor->flat<int64>(); variable
51 output_flat(i) = static_cast<int64>(bucket_id);
|
| string_to_hash_bucket_op.h | 45 auto output_flat = output_tensor->flat<int64>(); variable
54 output_flat(i) = static_cast<int64>(bucket_id);
87 auto output_flat = output_tensor->flat<int64>(); variable
96 output_flat(i) = static_cast<int64>(bucket_id);
|
| as_string_op.cc | 119 auto output_flat = output_tensor->flat<string>(); variable
125 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \
139 output_flat(i) = (input_flat(i)) ? "true" : "false";
145 output_flat(i) = strings::Printf(
152 output_flat(i) = strings::Printf(
|
| decode_compressed_op.cc | 93 auto output_flat = output_tensor->flat<string>(); variable
96 output_flat(i) = bytes_flat(i);
112 output_flat(i) = output_string;
|
| mfcc_op.cc | 77 float* output_flat = output_tensor->flat<float>().data(); variable
93 output_flat +
|
| sparse_softmax_op.cc | 81 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); variable
117 output_flat.data() + count, group_size);
|
| spectrogram_op.cc | 71 auto output_flat = output_tensor->flat<float>().data(); variable
76 output_flat + (channel * output_height * output_width);
|
| pack_op.cc | 104 auto output_flat = variable
117 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
123 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat);
127 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
|
| reduce_join_op.cc | 159 auto output_flat = output_tensor->flat<string>(); local
175 output_flat(output_index) =
|
| roll_op.cc | 293 auto output_flat = output->flat<T>().data(); variable
299 input_flat, output_flat, threshold, dim_range, isd);
303 output_flat, threshold, dim_range);
|
| stateful_random_ops.cc | 110 auto output_flat = output->flat<T>(); local
113 output_flat.size(), output_flat.data()));
166 auto output_flat = output->flat<T>(); variable
167 auto data = output_flat.data();
168 for (int64 i = 0; i < output_flat.size(); ++i) {
|
| concat_op.cc | 156 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); variable
159 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
165 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat);
169 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
|
| quantized_batch_norm_op.cc | 44 auto output_flat = output->flat<T2>(); local
81 output_flat(input_index) =
107 auto output_flat = output->flat<T2>(); local
156 output_flat(input_index) = output_value;
|
| quantized_concat_op.cc | 218 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); variable
222 &output_flat);
|
| random_op.cc | 203 auto output_flat = output->flat<T>(); variable
208 generator_.ReserveRandomOutputs(output_flat.size(), 256),
209 output_flat.data(), output_flat.size(), Distribution());
252 auto output_flat = output->flat<IntType>(); variable
257 generator_.ReserveRandomOutputs(output_flat.size(), 256),
258 output_flat.data(), output_flat.size(), dist);
[all...] |
| unicode_ops.cc | 301 auto output_flat = output_tensor->flat<string>(); variable
303 for (size_t i = 0; i < output_flat.size(); ++i) {
304 Transcode(&(output_flat(i)), input_encoder->converter_,
|
| /external/tensorflow/tensorflow/examples/adding_an_op/ |
| zero_out_op_kernel_2.cc | 72 auto output_flat = output->template flat<T>(); variable
77 output_flat(i) = T(0);
81 if (N > 0) output_flat(0) = input(0);
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| decode_video_op.cc | 90 auto output_flat = output->flat<uint8>(); variable
91 std::copy_n(output_data.begin(), output_data.size(), &output_flat(0));
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| audio_spectrogram.cc | 118 float* output_flat = GetTensorData<float>(output); local
123 output_flat + (channel * params->output_height * output_width);
|
| mfcc.cc | 116 float* output_flat = GetTensorData<float>(output); local
131 float* output_data = output_flat +
|
