Home
last modified time | relevance | path

Searched refs:GetDimensionSize (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcompute_constant_test.cc154 TEST_F(ComputeConstantTest, GetDimensionSize) { in TEST_F() argument
160 auto get_dimension_size = GetDimensionSize(add, 0); in TEST_F()
175 auto get_dimension_size = GetDimensionSize(add, 0); in TEST_F()
176 auto get_dimension_size_2 = GetDimensionSize(add, 0); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduction_ops.cc123 auto divisor = xla::GetDimensionSize(input, dimensions_to_reduce[0]); in BuildFinalizer()
125 auto size = xla::GetDimensionSize(input, dimensions_to_reduce[i]); in BuildFinalizer()
Dshape_op.cc105 size = xla::Mul(size, xla::GetDimensionSize(ctx->Input(0), i)); in Compile()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc387 TEST_F(XlaBuilderTest, GetDimensionSize) { in TEST_F() argument
390 GetDimensionSize(x, 1); in TEST_F()
Dxla_builder.h583 XlaOp GetDimensionSize(const XlaOp& operand, int64 dimension);
1006 friend XlaOp GetDimensionSize(const XlaOp& operand, int64 dimension);
1917 XlaOp GetDimensionSize(const XlaOp& operand, int64 dimension);
Dxla_builder.cc2494 XlaOp XlaBuilder::GetDimensionSize(const XlaOp& operand, int64 dimension) { in GetDimensionSize() function in xla::XlaBuilder
3517 XlaOp GetDimensionSize(const XlaOp& operand, int64 dimension) { in GetDimensionSize() function
3518 return operand.builder()->GetDimensionSize(operand, dimension); in GetDimensionSize()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1458 ## GetDimensionSize section in Operation Semantics
1461 [`XlaBuilder::GetDimensionSize`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla…
1466 <b> `GetDimensionSize(operand, dimension)` </b>
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc3029 TEST_F(HloEvaluatorTest, GetDimensionSize) { in TEST_F() argument