/external/tensorflow/tensorflow/core/kernels/ |
D | concat_lib_gpu.cc | 54 typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPUCall() 89 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPU()
|
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 | 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 | as_string_op.cc | 116 auto output_flat = output_tensor->flat<string>(); in Compute() local
|
D | mfcc_op.cc | 77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
|
D | spectrogram_op.cc | 71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local
|
D | sparse_softmax_op.cc | 78 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() 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 | 112 auto output_flat = output->flat_outer_dims<T>(); in Compute() local 282 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() local 489 auto output_flat = output->flat_outer_dims<T>(); in Compute() local 685 auto output_flat = output->flat_outer_dims<T>(); in Compute() local 1060 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 | 117 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
|
D | concat_op.cc | 134 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
|
D | mkl_concat_op.cc | 147 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local 225 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
|
D | roll_op.cc | 289 auto output_flat = output->flat<T>().data(); in Compute() local
|
D | random_op.cc | 203 auto output_flat = output->flat<T>(); in Compute() local 248 auto output_flat = output->flat<IntType>(); in Compute() local
|
D | tensor_array_ops.cc | 652 auto output_flat = in Compute() local 900 auto output_flat = in Compute() local
|
D | quantization_utils.h | 812 auto output_flat = output->flat<T3>(); in QuantizedAddUsingEigen() local 858 auto output_flat = output->flat<T3>(); in QuantizedAdd() local
|
D | batch_kernels.cc | 83 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Concat() 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
|