Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc140 const Tensor* concat_dim_tensor; in Compute() local
476 const Tensor& concat_dim_tensor = (AxisArgName == NAME_IS_CONCAT_DIM) in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_op.cc62 const Tensor& concat_dim_tensor = c->input(axis_input_index_); in Compute() local
H A Dquantized_concat_op.cc161 const Tensor* concat_dim_tensor = nullptr; in Compute() local