| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_simplifier.cc | 36 operand->concatenate_dimension() != concat->concatenate_dimension()) { in ConcatForwarding()
47 concat->shape(), new_operands, concat->concatenate_dimension())); in ConcatForwarding()
68 int64_t concat_dim = concat->concatenate_dimension(); in SliceConcatForwarding()
|
| D | algebraic_simplifier.cc | 1403 operand->concatenate_dimension() == in HandleConcatenate()
1404 concatenate->concatenate_dimension()) { in HandleConcatenate()
1416 concatenate->concatenate_dimension())); in HandleConcatenate()
1421 int64_t concatenate_dimension = concatenate->concatenate_dimension(); in HandleConcatenate() local
1431 int64_t slice_end = operands[i]->slice_limits(concatenate_dimension); in HandleConcatenate()
1438 operands[j]->slice_starts(concatenate_dimension) != slice_end) { in HandleConcatenate()
1447 if (k == concatenate_dimension) { in HandleConcatenate()
1458 slice_end = operands[j]->slice_limits(concatenate_dimension); in HandleConcatenate()
1464 concatenate_dimension, in HandleConcatenate()
1465 slice_end - operands[i]->slice_starts(concatenate_dimension)); in HandleConcatenate()
[all …]
|
| D | dynamic_dimension_inference.cc | 631 hlo->mutable_operand(i), {}, hlo->concatenate_dimension()); in HandleConcatenate()
635 hlo->operand(i)->shape().dimensions(hlo->concatenate_dimension()); in HandleConcatenate()
651 parent_->SetDynamicSize(hlo, {}, hlo->concatenate_dimension(), in HandleConcatenate()
659 int64_t concatenate_dimension = hlo->concatenate_dimension(); in HandleConcatenate() local
660 if (concatenate_dimension == dimension) { in HandleConcatenate()
|
| D | while_loop_concat_code_motion.cc | 475 concat->concatenate_dimension(), false)); in GroupHlosForConcat()
|
| D | layout_assignment.cc | 1601 if (sibling->shape().dimensions(user->concatenate_dimension()) > in PropagateOperandConstraint()
1602 operand->shape().dimensions(user->concatenate_dimension())) { in PropagateOperandConstraint()
|
| D | space_to_batch_converter.cc | 1572 if (consumer->concatenate_dimension() == old_batch_dim || in SupportedOpForPropagation()
1573 consumer->concatenate_dimension() == old_space_dim) { in SupportedOpForPropagation()
2745 DimLookUp(permute_dims, concat->concatenate_dimension()); in PropagateOnConcat()
|
| D | hlo_instruction.h | 1889 virtual int64_t concatenate_dimension() const;
|
| D | hlo_sharding_util.cc | 1501 indices->concatenate_dimension() == index_dim) { in GetGatherBatchParallelDims()
|
| D | dynamic_padder.cc | 1298 const int64_t concat_dim = concat->concatenate_dimension(); in RewriteDynamicConcat()
|
| D | hlo_instructions.h | 843 int64_t concatenate_dimension() const override { in concatenate_dimension() function
|
| D | hlo_instruction.cc | 4478 int64_t HloInstruction::concatenate_dimension() const { in concatenate_dimension() function in xla::HloInstruction
4479 return Cast<HloConcatenateInstruction>(this)->concatenate_dimension(); in concatenate_dimension()
|
| D | hlo_verifier.cc | 181 operand_shapes, concatenate->concatenate_dimension())); in HandleConcatenate()
|
| D | sharding_propagation.cc | 1340 const int64_t cdim = user.concatenate_dimension(); in GetShardingFromUser()
|
| D | hlo_instructions.cc | 1144 concatenate_dimension()); in CloneWithNewOperandsImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 240 int concatenate_dimension) { in Concat2D() argument
241 CHECK(0 <= concatenate_dimension && concatenate_dimension < 2); in Concat2D()
243 concatenate_dimension == 0 ? lhs.n1() + rhs.n1() : lhs.n1(), in Concat2D()
244 concatenate_dimension == 1 ? lhs.n2() + rhs.n2() : lhs.n2()); in Concat2D()
263 int concatenate_dimension) { in Concat3D() argument
264 CHECK(0 <= concatenate_dimension && concatenate_dimension < 3); in Concat3D()
269 if (i != concatenate_dimension) { in Concat3D()
298 int concatenate_dimension) { in Concat4D() argument
299 CHECK(0 <= concatenate_dimension && concatenate_dimension < 4); in Concat4D()
304 if (i != concatenate_dimension) { in Concat4D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | variadic_op_splitter.cc | 64 concat->concatenate_dimension()); in SplitConcatenate()
66 new_shape.set_dimensions(concat->concatenate_dimension(), in SplitConcatenate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1824 const int64_t dim = concat->concatenate_dimension(); in FindRotateRightPattern()
1862 const int64_t dim = concat->concatenate_dimension(); in FindPadWithWrapPattern()
|
| D | spmd_partitioner.cc | 2296 const int64_t dimension = hlo->concatenate_dimension(); in HandleConcatenate()
4725 const int64_t dim = hlo->concatenate_dimension(); in PreprocessHlos()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 969 builder_->getI64IntegerAttr(instruction->concatenate_dimension())) in ImportInstructionImpl()
|