| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu.cc | 40 typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPUCall()
75 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPU()
|
| D | regex_full_match_op.cc | 50 auto output_flat = output_tensor->flat<bool>(); in Compute() local
79 auto output_flat = output_tensor->flat<bool>(); in Compute() local
|
| D | string_to_hash_bucket_op.h | 45 auto output_flat = output_tensor->flat<int64>(); in Compute() local
87 auto output_flat = output_tensor->flat<int64>(); in Compute() local
|
| D | string_to_number_op.cc | 49 auto output_flat = output_tensor->flat<OutputType>(); in Compute() local
|
| D | unicode_script_op.cc | 35 auto output_flat = output_tensor->flat<int32>(); in Compute() local
|
| D | string_to_hash_bucket_op.cc | 42 auto output_flat = output_tensor->flat<int64>(); in Compute() local
|
| D | string_join_op.cc | 63 auto output_flat = output_tensor->flat<string>(); in Compute() local
|
| D | stateful_random_ops.cc | 110 auto output_flat = output->flat<T>(); in ComputeImpl() local
166 auto output_flat = output->flat<T>(); in Compute() local
|
| D | mfcc_op.cc | 77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
|
| D | as_string_op.cc | 119 auto output_flat = output_tensor->flat<string>(); in Compute() local
|
| D | spectrogram_op.cc | 71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local
|
| D | sparse_softmax_op.cc | 81 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local
|
| D | regex_replace_op.cc | 49 auto output_flat = output_tensor->flat<string>(); in InternalCompute() local
|
| D | pack_op.cc | 104 auto output_flat = in Compute() local
|
| D | decode_compressed_op.cc | 93 auto output_flat = output_tensor->flat<string>(); in Compute() local
|
| D | segment_reduction_ops.cc | 114 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
284 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() local
493 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
694 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
1069 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
|
| D | quantized_batch_norm_op.cc | 44 auto output_flat = output->flat<T2>(); in ReferenceBatchNorm() local
107 auto output_flat = output->flat<T2>(); in FixedPointBatchNorm() local
|
| D | list_kernels.h | 166 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
479 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
655 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
|
| D | concat_op.cc | 156 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
|
| D | reduce_join_op.cc | 159 auto output_flat = output_tensor->flat<string>(); in Compute() local
|
| D | quantized_concat_op.cc | 218 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
|
| /external/tensorflow/tensorflow/examples/adding_an_op/ |
| D | zero_out_op_kernel_2.cc | 72 auto output_flat = output->template flat<T>(); in Compute() local
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| D | decode_video_op.cc | 90 auto output_flat = output->flat<uint8>(); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | audio_spectrogram.cc | 118 float* output_flat = GetTensorData<float>(output); in Eval() local
|
| D | mfcc.cc | 116 float* output_flat = GetTensorData<float>(output); in Eval() local
|