Searched refs:all_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 255 std::vector<int64> all_dims(operand->shape().rank()); in MakeReduceHlo() local
256 std::iota(all_dims.begin(), all_dims.end(), 0); in MakeReduceHlo()
272 scalar_shape, operand, init_value, all_dims, reduce_computation)); in MakeReduceHlo()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 134 auto all_dims = Range(scope, Const(scope, 0), Rank(scope, grad_inputs[0]), in FillGrad() local
136 grad_outputs->push_back(ReduceSum(scope, grad_inputs[0], all_dims)); in FillGrad()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 171 auto all_dims = ops::Const(root.WithOpName("all_dims"), {0}, {1}); in TEST_F() local
173 auto all_valid = ops::All(root.WithOpName("all_valid"), valid, all_dims); in TEST_F()
|