Searched refs:operand_shapes_with_layout (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 136 absl::Span<const Shape> operand_shapes_with_layout, in BuildOpsSubmodule()
140 operand_shapes_with_layout, opaque, has_side_effect); in BuildOpsSubmodule()
|
| D | xla_client_test.py | 337 operand_shapes_with_layout=[
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1882 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall() argument
1894 if (operand_shapes_with_layout.has_value()) { in CustomCall()
1900 if (operands.size() != operand_shapes_with_layout->size()) { in CustomCall()
1904 operand_shapes_with_layout->size(), operands.size()); in CustomCall()
1907 for (const Shape& operand_shape : *operand_shapes_with_layout) { in CustomCall()
1918 operand_shapes_with_layout, has_side_effect, in CustomCall()
1926 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal() argument
1935 if (operand_shapes_with_layout.has_value()) { in CustomCallInternal()
1937 for (const Shape& operand_shape : *operand_shapes_with_layout) { in CustomCallInternal()
1961 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall() argument
[all …]
|
| D | xla_builder.h | 655 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout,
667 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout,
677 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout,
1232 absl::Span<const Shape> operand_shapes_with_layout, const string& opaque,
2054 absl::Span<const Shape> operand_shapes_with_layout,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 137 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout,
|
| D | mlir_hlo_builder.cc | 134 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal() argument
139 if (operand_shapes_with_layout.has_value()) in CustomCallInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 208 // 'operand_shapes_with_layout' must contain a shape with layout for each
211 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
|
| D | hlo_instructions.cc | 2395 absl::Span<const Shape> operand_shapes_with_layout) in HloCustomCallInstruction() argument
2402 operand_shapes_with_layout_(operand_shapes_with_layout.begin(), in HloCustomCallInstruction()
2403 operand_shapes_with_layout.end()), in HloCustomCallInstruction()
2589 cloned->operand_shapes_with_layout_ = operand_shapes_with_layout(); in CloneWithNewOperandsImpl()
|
| D | hlo_instructions.h | 1410 absl::Span<const Shape> operand_shapes_with_layout);
1487 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
|
| D | hlo_verifier.cc | 802 custom_call->operand_shapes_with_layout().size()); in HandleCustomCall()
805 custom_call->operand_shapes_with_layout()[i]; in HandleCustomCall()
|
| D | layout_assignment.cc | 488 custom_call->operand_shapes_with_layout()[i], custom_call, i)); in AddMandatoryConstraints()
696 custom_call->operand_shapes_with_layout()[i])); in CheckCustomCallLayout()
|
| D | hlo_instruction.h | 1008 absl::Span<const Shape> operand_shapes_with_layout, string opaque = "");
|
| D | hlo_instruction.cc | 556 proto.operand_shapes_with_layout()) { in CreateFromProto()
1599 absl::Span<const Shape> operand_shapes_with_layout, string opaque) { in CreateCustomCall() argument
1602 operand_shapes_with_layout); in CreateCustomCall()
|