Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h197 static StatusOr<Shape> InferSliceShape(const Shape& arg,
Dshape_inference_test.cc1273 ShapeInference::InferSliceShape(matrix_shape, {32, 0}, {64, 64}, {1, 1}); in TEST_F()
1282 ShapeInference::InferSliceShape(matrix_shape, {32, 0}, {33, 64}, {1, 1}); in TEST_F()
1292 ShapeInference::InferSliceShape(matrix_shape, {32, 0}, {64, 64}, {2, 4}); in TEST_F()
1301 ShapeInference::InferSliceShape(matrix_shape, {15, 0}, {20, 13}, {2, 4}); in TEST_F()
1310 ShapeInference::InferSliceShape(matrix_shape, {127, 0}, {129, 2}, {0, 1}); in TEST_F()
1319 ShapeInference::InferSliceShape(matrix_shape, {127, 0}, {129, 2}, {1, 1}); in TEST_F()
1328 ShapeInference::InferSliceShape(vector_shape, {2}, {4}, {1}); in TEST_F()
2354 ShapeInference::InferSliceShape(arg, {0}, {5}, {1}); in TEST_F()
Dhlo_creation_utils.cc86 TF_ASSIGN_OR_RETURN(Shape slice_shape, ShapeInference::InferSliceShape( in MakeSliceHlo()
Dhlo_verifier.cc840 ShapeInference::InferSliceShape( in HandleSlice()
Dshape_inference.cc2402 /* static */ StatusOr<Shape> ShapeInference::InferSliceShape( in InferSliceShape() function in xla::ShapeInference
Dhlo_evaluator_typed_visitor.h2439 ShapeInference::InferSliceShape(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc379 ShapeInference::InferSliceShape(new_backward_conv->shape(), start_indices, in CanonicalizeBackwardInputConvolution()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc957 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferSliceShape( in Slice()