Home
last modified time | relevance | path

Searched defs:GetDimensionSize (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/
H A Doperation_semantics.md1679 ## GetDimensionSize section in Operation Semantics
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc244 TEST_F(DynamismInferenceTest, GetDimensionSize) { in TEST_F() argument
577 TEST_F(UpperBoundInferenceTest, GetDimensionSize) { in TEST_F() argument
H A Dcompute_constant_test.cc153 TEST_F(ComputeConstantTest, GetDimensionSize) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder_test.cc498 TEST_F(XlaBuilderTest, GetDimensionSize) { in TEST_F() argument
H A Dxla_builder.cc3558 XlaOp XlaBuilder::GetDimensionSize(XlaOp operand, int64_t dimension) { in GetDimensionSize() function in xla::XlaBuilder
5004 XlaOp GetDimensionSize(const XlaOp operand, int64_t dimension) { in GetDimensionSize() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc4354 TEST_F(HloEvaluatorTest, GetDimensionSize) { in TEST_F() argument