Searched refs:operand_shapes_with_layout (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 199 // 'operand_shapes_with_layout' must contain a shape with layout for each
202 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
|
| D | hlo_instructions.cc | 2061 absl::Span<const Shape> operand_shapes_with_layout) in HloCustomCallInstruction() argument
2068 operand_shapes_with_layout_(operand_shapes_with_layout.begin(), in HloCustomCallInstruction()
2069 operand_shapes_with_layout.end()) { in HloCustomCallInstruction()
2183 cloned->operand_shapes_with_layout_ = operand_shapes_with_layout(); in CloneWithNewOperandsImpl()
|
| D | hlo_verifier.cc | 554 custom_call->operand_shapes_with_layout().size()); in HandleCustomCall()
557 custom_call->operand_shapes_with_layout()[i]; in HandleCustomCall()
|
| D | hlo_instructions.h | 1180 absl::Span<const Shape> operand_shapes_with_layout);
1219 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
|
| D | layout_assignment.cc | 474 custom_call->operand_shapes_with_layout()[i], custom_call, i)); in AddMandatoryConstraints()
679 custom_call->operand_shapes_with_layout()[i])); in CheckCustomCallLayout()
|
| D | hlo_instruction.cc | 450 proto.operand_shapes_with_layout()) { in CreateFromProto()
1290 absl::Span<const Shape> operand_shapes_with_layout, in CreateCustomCall() argument
1293 shape, operands, custom_call_target, opaque, operand_shapes_with_layout); in CreateCustomCall()
|
| D | hlo_instruction.h | 776 absl::Span<const Shape> operand_shapes_with_layout,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 621 const std::vector<Shape>& operand_shapes_with_layout, in CustomCall() argument
630 operand_shapes_with_layout, opaque); in CustomCall()
|
| D | local_computation_builder.h | 314 const std::vector<Shape>& operand_shapes_with_layout,
|
| D | xla_client.py | 1422 operand_shapes_with_layout, argument
1441 operand_shapes_with_layout, opaque)
|
| D | xla_client_test.py | 323 operand_shapes_with_layout=(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1496 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout) { in CustomCall() argument
1508 if (operand_shapes_with_layout.has_value()) { in CustomCall()
1514 if (operands.size() != operand_shapes_with_layout->size()) { in CustomCall()
1518 operand_shapes_with_layout->size(), operands.size()); in CustomCall()
1522 for (const Shape& operand_shape : *operand_shapes_with_layout) { in CustomCall()
3090 absl::Span<const Shape> operand_shapes_with_layout, in CustomCallWithLayout() argument
3093 operand_shapes_with_layout); in CustomCallWithLayout()
|
| D | xla_builder.h | 438 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout);
830 absl::Span<const Shape> operand_shapes_with_layout, const string& opaque);
1443 absl::Span<const Shape> operand_shapes_with_layout,
|