Searched refs:zero_base (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 1053 const int64_t zero_base[] = {0, 0, 0, 0}; in TEST_F() local
1060 ShapeUtil::ForEachIndex(source.shape(), zero_base, dimensions, step, in TEST_F()
1084 ShapeUtil::ForEachIndex(source.shape(), zero_base, copy_size, step, in TEST_F()
1265 std::vector<int64_t> zero_base(data.dimensions.size(), 0); in TEST_F() local
1273 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
1308 std::vector<int64_t> zero_base(data.dimensions.size(), 0); in TEST_F() local
1316 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
|
| D | shape_util_test.cc | 598 std::vector<int64_t> zero_base(data.dimensions.size(), 0); in TEST() local
601 ShapeUtil::ForEachIndex(shape, zero_base, data.dimensions, step, in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1404 std::vector<int64_t> zero_base(evaluated_operand.shape().dimensions_size(),
1408 ShapeUtil::ForEachIndex(evaluated_operand.shape(), zero_base,
|
| D | hlo_evaluator.cc | 3478 std::vector<int64_t> zero_base(rank, 0); in HandleSort() local
3490 key_shape, zero_base, key_shape.dimensions(), increment, in HandleSort()
|