| /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 | 54 auto output_flat = output_tensor->flat<bool>(); in Compute() local
107 auto output_flat = output_tensor->flat<bool>(); in Compute() local
|
| D | segment_reduction_ops_impl.h | 123 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
295 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() 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_fast_op.h | 45 auto output_flat = output_tensor->flat<int64>(); in Compute() local
|
| D | string_to_hash_bucket_op.h | 52 auto output_flat = output_tensor->flat<int64>(); in Compute() local
|
| D | string_to_hash_bucket_op.cc | 41 auto output_flat = output_tensor->flat<int64>(); in Compute() local
|
| D | string_join_op.cc | 63 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
|
| D | mfcc_op.cc | 77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
|
| D | stateful_random_ops.cc | 137 auto output_flat = output->flat<T>(); in StatefulRandomCompute() local
290 auto output_flat = output->flat<T>(); in Compute() local
333 auto output_flat = output->flat<T>(); in Compute() local
|
| D | as_string_op.cc | 141 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
|
| D | spectrogram_op.cc | 71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local
|
| D | sparse_softmax_op.cc | 77 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local
|
| D | pack_op.cc | 91 auto output_flat = in Compute() local
|
| D | random_op.cc | 75 auto output_flat = output->flat<T>(); in Compute() local
124 auto output_flat = output->flat<IntType>(); in Compute() local
|
| D | decode_compressed_op.cc | 93 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
|
| D | regex_replace_op.cc | 51 auto output_flat = output_tensor->flat<tstring>(); in InternalCompute() 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 | 145 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
456 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
632 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
|
| D | unsorted_segment_join_op.cc | 118 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
|
| D | sobol_op.cc | 159 auto output_flat = output->flat<T>(); in Compute() local
|
| /external/tensorflow/tensorflow/lite/python/testdata/ |
| D | double_op.cc | 42 auto output_flat = output_tensor->flat<int32>(); 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/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_detokenizer_op.cc | 59 auto output_flat = output_tensor->flat<tensorflow::tstring>(); in Compute() local
|