Home
last modified time | relevance | path

Searched defs:output_flat (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu.cc54 typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPUCall()
89 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPU()
Dstring_to_hash_bucket_op.h45 auto output_flat = output_tensor->flat<int64>(); in Compute() local
87 auto output_flat = output_tensor->flat<int64>(); in Compute() local
Dstring_to_number_op.cc49 auto output_flat = output_tensor->flat<OutputType>(); in Compute() local
Dstring_to_hash_bucket_op.cc42 auto output_flat = output_tensor->flat<int64>(); in Compute() local
Dstring_join_op.cc63 auto output_flat = output_tensor->flat<string>(); in Compute() local
Das_string_op.cc116 auto output_flat = output_tensor->flat<string>(); in Compute() local
Dmfcc_op.cc77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
Dspectrogram_op.cc71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local
Dsparse_softmax_op.cc78 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local
Dpack_op.cc104 auto output_flat = in Compute() local
Ddecode_compressed_op.cc93 auto output_flat = output_tensor->flat<string>(); in Compute() local
Dsegment_reduction_ops.cc112 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
Dquantized_batch_norm_op.cc44 auto output_flat = output->flat<T2>(); in ReferenceBatchNorm() local
107 auto output_flat = output->flat<T2>(); in FixedPointBatchNorm() local
Dlist_kernels.h117 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
Dconcat_op.cc134 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
Dreduce_join_op.cc159 auto output_flat = output_tensor->flat<string>(); in Compute() local
Dquantized_concat_op.cc218 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
Dmkl_concat_op.cc147 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
Droll_op.cc289 auto output_flat = output->flat<T>().data(); in Compute() local
Drandom_op.cc203 auto output_flat = output->flat<T>(); in Compute() local
248 auto output_flat = output->flat<IntType>(); in Compute() local
Dtensor_array_ops.cc652 auto output_flat = in Compute() local
900 auto output_flat = in Compute() local
Dquantization_utils.h812 auto output_flat = output->flat<T3>(); in QuantizedAddUsingEigen() local
858 auto output_flat = output->flat<T3>(); in QuantizedAdd() local
Dbatch_kernels.cc83 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Concat() local
/external/tensorflow/tensorflow/examples/adding_an_op/
Dzero_out_op_kernel_2.cc72 auto output_flat = output->template flat<T>(); in Compute() local
/external/tensorflow/tensorflow/contrib/ffmpeg/
Ddecode_video_op.cc90 auto output_flat = output->flat<uint8>(); in Compute() local