Searched refs:MakeReshapeHlo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 107 StatusOr<HloInstruction*> MakeReshapeHlo(const Shape& result_shape, in MakeReshapeHlo() function
114 StatusOr<HloInstruction*> MakeReshapeHlo( in MakeReshapeHlo() function
118 return MakeReshapeHlo(new_shape, operand); in MakeReshapeHlo()
332 return MakeReshapeHlo(output_shape, operand); in CollapseFirstNDims()
343 return MakeReshapeHlo(new_shape_dims, operand); in PrependDegenerateDims()
360 return MakeReshapeHlo(new_shape, operand); in ExpandFirstDimIntoNDims()
385 return MakeReshapeHlo(output_shape, operand); in ElideDegenerateDims()
417 return MakeReshapeHlo(output_shape, operand); in InsertDegenerateDims()
|
| D | hlo_creation_utils.h | 69 StatusOr<HloInstruction*> MakeReshapeHlo(const Shape& result_shape,
72 StatusOr<HloInstruction*> MakeReshapeHlo(
|
| D | batch_dot_simplification.cc | 77 MakeReshapeHlo(batch_dot->shape(), new_dot)); in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | scatter_expander.cc | 67 MakeReshapeHlo(new_shape, scatter_indices)); in CanonicalizeScatterIndices()
|