Home
last modified time | relevance | path

Searched defs:output_dim1 (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_op.cc160 int64_t output_dim1 = output->NumElements() / inputs_flat_dim0; in Compute() local
H A Dquantized_concat_op.cc216 int64_t output_dim1 = output->NumElements() / inputs_flat_dim0; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc215 int64 output_dim1 = output->NumElements() / inputs_flat_dim0; in Compute() local