Searched refs:WithSubshape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | pattern_matcher_test.cc | 82 match::Shape().WithSubshape({0}, match::Shape()).WithElementType(F32))); in TEST_F()
96 Match(&array_shape, match::Shape().WithSubshape({0}, match::Shape()))); in TEST_F()
116 match::Shape().WithSubshape( in TEST_F()
122 match::Shape().WithSubshape( in TEST_F()
126 match::Shape().WithSubshape( in TEST_F()
132 match::Shape().WithSubshape( in TEST_F()
138 match::Shape().WithSubshape( in TEST_F()
142 match::Shape().WithSubshape( in TEST_F()
147 Match(&tuple_shape, match::Shape().WithSubshape({2}, match::Shape()))); in TEST_F()
149 Match(&tuple_shape, match::Shape().WithSubshape({0, 0}, match::Shape()))); in TEST_F()
[all …]
|
| D | pattern_matcher.h | 1091 auto WithSubshape(
1105 return WithSubshape(index,
1119 return WithSubshape(index,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.cc | 145 HloProtoEvaluator& WithSubshape(ShapeIndex shape_index) { in WithSubshape() function
615 .WithSubshape(context.shape_index) in AnalyzeConstantValueFallback()
990 .WithSubshape(context.shape_index) in AnalyzeConstant()
1348 .WithSubshape(context.shape_index) in AnalyzeIsDynamic()
|