Searched refs:set_dynamic_dimension (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 88 array_shape.set_dynamic_dimension(2, false); in TEST_F()
119 dynamic_matrix.set_dynamic_dimension(1, true); in TEST_F()
125 ->set_dynamic_dimension(1, true); in TEST_F()
131 dynamic_matrix.set_dynamic_dimension(1, true); in TEST_F()
138 ->set_dynamic_dimension(1, true); in TEST_F()
|
| D | shape.h | 71 void set_dynamic_dimension(int dimension, bool is_dynamic) { in set_dynamic_dimension() function
|
| D | shape_util.cc | 184 shape.set_dynamic_dimension(i, dynamic_dimensions[i]); in MakeValidatedShape()
938 new_shape.set_dynamic_dimension(permutation[i], in PermuteDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 854 output_shape.set_dynamic_dimension(dimension_to_match, small_is_dynamic); in InferInDimBroadcastShape()
2711 inferred_shape.set_dynamic_dimension(unmodified.second, true); in InferReshapeShape()
2799 inferred_shape.set_dynamic_dimension(i, true); in InferSelectShape()
|
| D | hlo_parser.cc | 3559 result->set_dynamic_dimension(i, dynamic_dimensions[i]); in ParseShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 686 output_shape.set_dynamic_dimension(broadcast_dimensions[i], in BroadcastInDim()
|