Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu.cc40 typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPUCall()
75 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPU()
H A Dregex_full_match_op.cc54 auto output_flat = output_tensor->flat<bool>(); in Compute() local
107 auto output_flat = output_tensor->flat<bool>(); in Compute() local
H A Dstateful_random_ops.cc137 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
H A Dsegment_reduction_ops_impl.h126 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
327 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() local
H A Drandom_op.cc75 auto output_flat = output->flat<T>(); in Compute() local
124 auto output_flat = output->flat<IntType>(); in Compute() local
H A Dlist_kernels.h249 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
560 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
743 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
H A Dquantized_batch_norm_op.cc44 auto output_flat = output->flat<T2>(); in ReferenceBatchNorm() local
107 auto output_flat = output->flat<T2>(); in FixedPointBatchNorm() local
H A Dstring_to_number_op.cc49 auto output_flat = output_tensor->flat<OutputType>(); in Compute() local
H A Dunicode_script_op.cc35 auto output_flat = output_tensor->flat<int32>(); in Compute() local
H A Dstring_to_hash_bucket_fast_op.h45 auto output_flat = output_tensor->flat<int64_t>(); in Compute() local
H A Dstring_to_hash_bucket_op.h52 auto output_flat = output_tensor->flat<int64_t>(); in Compute() local
H A Dstring_to_hash_bucket_op.cc41 auto output_flat = output_tensor->flat<int64_t>(); in Compute() local
H A Dstring_join_op.cc63 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
H A Dtensor_to_hash_bucket_op.cc53 auto output_flat = output_tensor->flat<int64_t>(); in Compute() local
H A Dmfcc_op.cc77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
H A Dspectrogram_op.cc71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local
H A Dpack_op.cc90 auto output_flat = output->shaped<T, 2>({before_dim, after_dim * axis_dim}); in Compute() local
H A Dsparse_softmax_op.cc80 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local
H A Das_string_op.cc153 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
H A Ddecode_compressed_op.cc93 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
H A Dtensor_array_ops.cc678 auto output_flat = in Compute() local
922 auto output_flat = in Compute() local
H A Dregex_replace_op.cc51 auto output_flat = output_tensor->flat<tstring>(); in InternalCompute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/testdata/
H A Ddouble_op.cc44 auto output_flat = output_tensor->flat<T>(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/adding_an_op/
H A Dzero_out_op_kernel_2.cc72 auto output_flat = output->template flat<T>(); in Compute() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
H A Dsentencepiece_detokenizer_op.cc59 auto output_flat = output_tensor->flat<tensorflow::tstring>(); in Compute() local

12