Searched refs:zero_base (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 977 const int64 zero_base[] = {0, 0, 0, 0}; in TEST_F() local
984 ShapeUtil::ForEachIndex(source.shape(), zero_base, dimensions, step, in TEST_F()
1008 ShapeUtil::ForEachIndex(source.shape(), zero_base, copy_size, step, in TEST_F()
1192 std::vector<int64> zero_base(data.dimensions.size(), 0); in TEST_F() local
1200 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
1234 std::vector<int64> zero_base(data.dimensions.size(), 0); in TEST_F() local
1242 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
|
| D | shape_util_test.cc | 559 std::vector<int64> zero_base(data.dimensions.size(), 0); in TEST() local
562 ShapeUtil::ForEachIndex(shape, zero_base, data.dimensions, step, in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1458 std::vector<int64> zero_base(rank, 0); in HandleSort() local
1466 key_shape, zero_base, AsInt64Slice(key_shape.dimensions()), increment, in HandleSort()
|
| D | hlo_evaluator_typed_visitor.h | 1444 std::vector<int64> zero_base(evaluated_operand.shape().dimensions_size(), in HandlePad()
1449 evaluated_operand.shape(), zero_base, in HandlePad()
|