Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_concat_op_test.py32 val_shape=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize.cc462 const auto val_shape = val_type.getShape(); in IsLastDimensionEqualOne() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc1664 Shape val_shape = ShapeUtil::MakeShape(S32, {1}); in TEST_F() local
1675 Shape val_shape = ShapeUtil::MakeShape(U32, {}); in TEST_F() local
H A Dshape_inference.cc2430 const Shape& shape, const Shape& val_shape, int64_t dimension) { in InferSetDimensionSizeShape()