Searched refs:CollapseFirstNDims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 55 CollapseFirstNDims(param, 1)); in TEST_F()
74 CollapseFirstNDims(param, 2)); in TEST_F()
|
| D | scatter_expander.cc | 85 return CollapseFirstNDims( in CanonicalizeScatterIndices()
127 return CollapseFirstNDims(updates, num_scatter_dims); in AdjustScatterDims()
|
| D | hlo_creation_utils.h | 166 StatusOr<HloInstruction*> CollapseFirstNDims(HloInstruction* operand, int64 n);
|
| D | gather_expander.cc | 78 return CollapseFirstNDims( in CanonicalizeGatherIndices()
|
| D | hlo_creation_utils.cc | 311 StatusOr<HloInstruction*> CollapseFirstNDims(HloInstruction* operand, int64 n) { in CollapseFirstNDims() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 869 Shape CollapseFirstNDims(const Shape& shape, int64 n) { in CollapseFirstNDims() function
882 llvm_ir::IrArray CollapseFirstNDims(llvm::IRBuilder<>* b, in CollapseFirstNDims() function
889 Shape new_shape = CollapseFirstNDims(shape, n); in CollapseFirstNDims()
947 CollapseFirstNDims(b, lhs_array, num_batch_dims); in EmitBatchDotOperation()
949 CollapseFirstNDims(b, rhs_array, num_batch_dims); in EmitBatchDotOperation()
951 CollapseFirstNDims(b, target_array, num_batch_dims); in EmitBatchDotOperation()
|