Searched refs:InferSliceShape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 178 static StatusOr<Shape> InferSliceShape(const Shape& arg,
|
| D | shape_inference_test.cc | 836 ShapeInference::InferSliceShape(matrix_shape, {32, 0}, {64, 64}, {1, 1}); in TEST_F()
845 ShapeInference::InferSliceShape(matrix_shape, {32, 0}, {64, 64}, {2, 4}); in TEST_F()
854 ShapeInference::InferSliceShape(matrix_shape, {15, 0}, {20, 13}, {2, 4}); in TEST_F()
863 ShapeInference::InferSliceShape(matrix_shape, {127, 0}, {129, 2}, {0, 1}); in TEST_F()
872 ShapeInference::InferSliceShape(matrix_shape, {127, 0}, {129, 2}, {1, 1}); in TEST_F()
881 ShapeInference::InferSliceShape(vector_shape, {2}, {4}, {1}); in TEST_F()
1705 ShapeInference::InferSliceShape(arg, {0}, {5}, {1}); in TEST_F()
|
| D | hlo_creation_utils.cc | 75 TF_ASSIGN_OR_RETURN(Shape slice_shape, ShapeInference::InferSliceShape( in MakeSliceHlo()
|
| D | hlo_verifier.cc | 570 ShapeInference::InferSliceShape( in HandleSlice()
|
| D | shape_inference.cc | 2203 /* static */ StatusOr<Shape> ShapeInference::InferSliceShape( in InferSliceShape() function in xla::ShapeInference
|
| D | hlo_evaluator_typed_visitor.h | 2418 ShapeInference::InferSliceShape( in HandleSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 371 ShapeInference::InferSliceShape(new_backward_conv->shape(), start_indices, in CanonicalizeBackwardInputConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 729 Shape shape, ShapeInference::InferSliceShape( in Slice()
|