Home
last modified time | relevance | path

Searched defs:concat_dim_size (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dmisc.cpp569 const std::vector<ExprHandle>& dims) { in computeCatWoConditionals()
601 ExprPtr concat_dim_size = nullptr; in computeCatWoConditionals() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc889 int64* concat_dim_size) { in FindMklCommonFormat()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc849 int64_t concat_dim_size = 0; in ExchangeHalo() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc1237 uint32_t concat_dim_size = 0; in TransformPackIntoSupportedOps() local