Searched refs:dimensions_to_reduce (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 148 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce,
|
| D | ir_emitter_unnested.cc | 1575 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce, in EmitReductionToVector() argument
1588 if (std::find(dimensions_to_reduce.begin(), dimensions_to_reduce.end(), in EmitReductionToVector()
1589 input_dim) == dimensions_to_reduce.end()) { in EmitReductionToVector()
1663 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce(reduce->dimensions()); in HandleReduce() local
1689 dimensions_to_reduce, reducer); in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 472 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce) { in Reduce() argument
474 dimensions_to_reduce); in Reduce()
|
| D | local_computation_builder.h | 239 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 135 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce,
|
| D | shape_inference.cc | 1847 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce, in InferReduceShape() argument
1850 for (int64 dimension : dimensions_to_reduce) { in InferReduceShape()
1860 std::set<int64> dimensions_to_reduce_set(dimensions_to_reduce.begin(), in InferReduceShape()
1861 dimensions_to_reduce.end()); in InferReduceShape()
|
| D | hlo_instruction.h | 370 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce,
|
| D | shape_inference_test.cc | 62 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce) { in ExpectInferredReduceShape() argument
65 arg, f32_, dimensions_to_reduce, to_apply); in ExpectInferredReduceShape()
|
| D | hlo_instruction.cc | 573 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce, in CreateReduce() argument
578 instruction->dimensions_.assign(dimensions_to_reduce.begin(), in CreateReduce()
579 dimensions_to_reduce.end()); in CreateReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| D | hlo_parser.cc | 754 optional<std::vector<int64>> dimensions_to_reduce; in ParseInstruction() local
756 &dimensions_to_reduce}; in ParseInstruction()
763 *dimensions_to_reduce, *reduce_computation)); in ParseInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | computation_builder.cc | 1272 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce) { in Reduce() argument
1277 for (int64 dimension : dimensions_to_reduce) { in Reduce()
|
| D | computation_builder.h | 533 tensorflow::gtl::ArraySlice<int64> dimensions_to_reduce);
|