Searched refs:zero_base (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 1046 const int64 zero_base[] = {0, 0, 0, 0}; in TEST_F() local
1053 ShapeUtil::ForEachIndex(source.shape(), zero_base, dimensions, step, in TEST_F()
1077 ShapeUtil::ForEachIndex(source.shape(), zero_base, copy_size, step, in TEST_F()
1261 std::vector<int64> zero_base(data.dimensions.size(), 0); in TEST_F() local
1269 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
1303 std::vector<int64> zero_base(data.dimensions.size(), 0); in TEST_F() local
1311 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
|
| D | shape_util_test.cc | 583 std::vector<int64> zero_base(data.dimensions.size(), 0); in TEST() local
586 ShapeUtil::ForEachIndex(shape, zero_base, data.dimensions, step, in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 2178 std::vector<int64> zero_base(rank, 0); in HandleSort() local
2186 key_shape, zero_base, AsInt64Slice(key_shape.dimensions()), increment, in HandleSort()
|
| D | hlo_evaluator_typed_visitor.h | 1588 std::vector<int64> zero_base(evaluated_operand.shape().dimensions_size(),
1593 evaluated_operand.shape(), zero_base,
|