| /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)),
|
| 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 | 116 auto output_flat = output_tensor->flat<string>(); variable
122 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \
135 output_flat(i) = (input_flat(i)) ? "true" : "false";
141 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 | 78 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); variable
114 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 | 289 auto output_flat = output->flat<T>().data(); variable
295 input_flat, output_flat, threshold, dim_range, isd);
299 output_flat, threshold, dim_range);
|
| concat_op.cc | 134 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); variable
137 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
143 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat);
147 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
|
| list_kernels.h | 117 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); variable
121 ConcatGPU<T>(c, inputs_flat, output, &output_flat);
125 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());
248 auto output_flat = output->flat<IntType>(); variable
253 generator_.ReserveRandomOutputs(output_flat.size(), 256),
254 output_flat.data(), output_flat.size(), dist);
|
| mkl_concat_op.cc | 147 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); local
148 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
225 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1});
226 ConcatCPU<T>(c->device(), inputs_flat, &output_flat);
[all...] |
| segment_reduction_ops.cc | 112 auto output_flat = output->flat_outer_dims<T>(); variable
162 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape);
166 T* out_slice_ptr = &output_flat(out_index, 0);
282 auto output_flat = output->flat_outer_dims<T>();
287 output_flat);
489 auto output_flat = output->flat_outer_dims<T>(); variable
492 data.NumElements(), data_ptr, output_flat);
685 auto output_flat = output->flat_outer_dims<T>(); variable
720 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape);
724 auto out = output_flat.template chip<0>(out_index)
1060 auto output_flat = output->flat_outer_dims<T>(); variable
[all...] |
| tensor_array_ops.cc | 652 auto output_flat = variable
673 ConcatGPU<T>(ctx, input_tensors_flat, output_tensor, &output_flat);
677 ConcatCPU<T>(ctx->device(), input_tensors_flat, &output_flat);
900 auto output_flat = variable
[all...] |
| batch_kernels.cc | 83 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); local
86 ConcatGPU<T>(context, inputs_flat, output, &output_flat);
90 ConcatCPU<T>(context->device(), inputs_flat, &output_flat);
[all...] |
| /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));
|
