Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_utils_test.cc363 constexpr int kNumDims = 3; in TEST_P() local
381 constexpr int kNumDims = 3; in TEST_P() local
399 constexpr int kNumDims = 3; in TEST_P() local
417 constexpr int kNumDims = 3; in TEST_P() local
H A Dsparse_dense_binary_op_shared_test.cc240 const int kNumDims = 3; in MakeSparseTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h1242 const int kNumDims = dim.size(); in CreateBlockedMemDescHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc2791 const int kNumDims = 5; in ConvertConv3DHelper() local
2987 const int kNumDims = 5; in ConvertPool3D() local