Searched refs:inferred_dimension (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 282 int64_t inferred_dimension);
|
| D | dynamic_dimension_inference.cc | 934 if (hlo->inferred_dimension() != -1) { in HandleReshape()
992 hlo->shape().dimensions(hlo->inferred_dimension()); in HandleReshape()
999 parent_->SetDynamicSize(hlo, {}, hlo->inferred_dimension(), dynamic_size); in HandleReshape()
1127 output_dynamic_dimension = reshape->inferred_dimension(); in HandleReshape()
|
| D | hlo_instructions.cc | 1291 int64_t inferred_dimension) in HloReshapeInstruction() argument
1293 inferred_dimension_(inferred_dimension) { in HloReshapeInstruction()
1307 if (inferred_dimension() == -1) { in ExtraAttributesToStringImpl()
1310 return {StrCat("inferred_dimension=", inferred_dimension())}; in ExtraAttributesToStringImpl()
1318 return inferred_dimension() == casted_other.inferred_dimension(); in IdenticalSlowPath()
1326 inferred_dimension()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.cc | 922 int64_t inferred_dimension = -1; in CreateFromProto() local
924 inferred_dimension = proto.dimensions()[0]; in CreateFromProto()
931 instruction = CreateReshape(shape, operands(0), inferred_dimension); in CreateFromProto()
1746 const Shape& shape, HloInstruction* operand, int64_t inferred_dimension) { in CreateReshape() argument
1753 inferred_dimension); in CreateReshape()
4489 int64_t HloInstruction::inferred_dimension() const { in inferred_dimension() function in xla::HloInstruction
4490 return Cast<HloReshapeInstruction>(this)->inferred_dimension(); in inferred_dimension()
|
| D | hlo_instruction.h | 1013 int64_t inferred_dimension = -1);
1895 int64_t inferred_dimension() const;
|
| D | hlo_instructions.h | 990 int64_t inferred_dimension);
991 int64_t inferred_dimension() const { return inferred_dimension_; } in inferred_dimension() function
|
| D | shape_inference.cc | 3085 absl::Span<const int64_t> new_sizes, int64_t inferred_dimension) { in InferReshapeShape() argument
3212 output_dynamic_dimension = inferred_dimension; in InferReshapeShape()
|
| D | hlo_parser.cc | 1737 optional<int64_t> inferred_dimension; in CreateInstruction() local
1739 &inferred_dimension}; in CreateInstruction()
1746 *shape, operands[0], inferred_dimension.value_or(-1))); in CreateInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 212 int64_t inferred_dimension) override;
|
| D | mlir_hlo_builder.cc | 486 int64_t inferred_dimension) { in ReshapeInternal() argument
489 if (inferred_dimension != -1) in ReshapeInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1007 int64_t inferred_dimension) { in ReshapeInternal() argument
1012 if (inferred_dimension != -1) { in ReshapeInternal()
1013 instr.add_dimensions(inferred_dimension); in ReshapeInternal()
1184 int64_t inferred_dimension) { in Reshape() argument
1189 new_sizes, inferred_dimension)); in Reshape()
1193 return ReshapeInternal(shape, transposed, inferred_dimension); in Reshape()
1198 int64_t inferred_dimension) { in Reshape() argument
1203 return Reshape(operand, dimensions, new_sizes, inferred_dimension); in Reshape()
1208 int64_t inferred_dimension) { in Reshape() argument
1210 return ReshapeInternal(shape, operand, inferred_dimension); in Reshape()
[all …]
|
| D | xla_builder.h | 473 int64_t inferred_dimension = -1);
476 int64_t inferred_dimension = -1);
479 int64_t inferred_dimension = -1);
1002 int64_t inferred_dimension);
1130 int64_t inferred_dimension);
1811 int64_t inferred_dimension);
|