| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 47 optional<int64> operand_idx = ProfitableToMakeDotOperandColumnMajor(*user); in ShouldMakeAllUsersColMajor() local
48 if (!operand_idx || user->operand(*operand_idx) != instruction || in ShouldMakeAllUsersColMajor()
58 optional<int64> operand_idx = in ShouldMakeOperandColumnMajor() local
60 if (!operand_idx) { in ShouldMakeOperandColumnMajor()
64 const HloInstruction* operand = instruction.operand(*operand_idx); in ShouldMakeOperandColumnMajor()
77 return it->second ? operand_idx : nullopt; in ShouldMakeOperandColumnMajor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_propagation.cc | 30 for (int operand_idx = 0; operand_idx < instruction->operand_count(); in Run() local
31 ++operand_idx) { in Run()
34 instruction->operand(operand_idx)->shape())) { in Run()
37 instruction->fused_parameter(operand_idx), in Run()
|
| D | bfloat16_normalization.cc | 80 Status InsertConvertBeforeOperand(HloInstruction* hlo, int64 operand_idx,
219 HloInstruction* hlo, int64 operand_idx, PrimitiveType from, in InsertConvertBeforeOperand() argument
221 auto operand = hlo->mutable_operand(operand_idx); in InsertConvertBeforeOperand()
228 hlo->ReplaceOperandWithDifferentShape(operand_idx, new_operand)); in InsertConvertBeforeOperand()
|
| D | hlo_instruction_test.cc | 1137 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
1138 ++operand_idx) { in TEST_F()
1139 const HloInstruction* operand = fusion->operand(operand_idx); in TEST_F()
1141 EXPECT_FALSE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1143 EXPECT_TRUE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1178 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
1179 ++operand_idx) { in TEST_F()
1180 if (fusion->operand(operand_idx) == y) { in TEST_F()
1181 EXPECT_FALSE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1183 EXPECT_TRUE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
|
| D | elemental_ir_emitter.cc | 1835 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in EmitElementalConcatenate() local
1836 ++operand_idx) { in EmitElementalConcatenate()
1837 const HloInstruction* operand = hlo->operand(operand_idx); in EmitElementalConcatenate()
1839 exit_block, StrCat("concat_index_not_from_operand", operand_idx), b_); in EmitElementalConcatenate()
2688 for (int64 operand_idx = 0; operand_idx < input_count; ++operand_idx) { in EmitElementalReduceWindow() local
2690 input_generators[operand_idx](input_index)); in EmitElementalReduceWindow()
2691 input_values[input_count + operand_idx] = input_value; in EmitElementalReduceWindow()
2692 input_values[operand_idx] = Load(accum_ptrs[operand_idx]); in EmitElementalReduceWindow()
2698 for (int64 operand_idx = 0; operand_idx < accum_values.size(); in EmitElementalReduceWindow() local
2699 ++operand_idx) { in EmitElementalReduceWindow()
[all …]
|
| D | hlo_instructions.cc | 1163 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
1280 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
1412 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
1413 if (!operand_idx.has_value()) { in IsElementwiseImpl()
1425 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1426 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1886 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
|
| D | hlo_instructions.h | 771 const absl::optional<int64>& operand_idx) const override;
854 const absl::optional<int64>& operand_idx) const override;
997 const absl::optional<int64>& operand_idx) const override;
1027 const absl::optional<int64>& operand_idx) const override;
|
| D | hlo_instruction.h | 1483 bool IsElementwiseOnOperand(int64 operand_idx) const;
2020 const absl::optional<int64>& operand_idx) const;
|
| D | hlo_instruction.cc | 2673 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
2675 return operand_idx.has_value() && operand_idx.value() == 0; in IsElementwiseImpl()
3455 bool HloInstruction::IsElementwiseOnOperand(int64 operand_idx) const { in IsElementwiseOnOperand()
3456 return IsElementwiseImpl(operand_idx); in IsElementwiseOnOperand()
|
| D | hlo_verifier.cc | 567 int64 operand_idx = parameter_idx / 2; in HandleSort() local
569 sort->operand(operand_idx)->shape().element_type(), {}); in HandleSort()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 143 for (int operand_idx : llvm::seq<int>(0, op->getNumOperands())) { in VerifyControlOperandsAfterAllData() local
144 if (op->getOperand(operand_idx).getType().isa<ControlType>()) { in VerifyControlOperandsAfterAllData()
149 return op->emitOpError() << "found non-control operand #" << operand_idx in VerifyControlOperandsAfterAllData()
295 for (int operand_idx : llvm::seq<int>(0, yield.getNumOperands())) { in Verify() local
296 if (island.getResult(operand_idx).getType() != in Verify()
297 yield.getOperand(operand_idx).getType()) in Verify()
299 << "operand #" << operand_idx << " type mismatch island results"; in Verify()
304 for (int operand_idx : llvm::seq<int>(0, island.getNumResults() - 1)) { in Verify() local
305 if (island.getResult(operand_idx).getType() == control_type) in Verify()
307 << "unexpected control type for operand #" << operand_idx; in Verify()
|
| D | tf_device.cc | 505 int32_t operand_idx = operand_type_and_idx.index(); in Verify() local
506 for (int32_t i = n * operand_idx, e = i + n; i < e; ++i) in Verify()
509 << " and terminator operand " << operand_idx; in Verify()
|
| D | tf_ops_a_m.cc | 1110 unsigned operand_idx = 0; in matchAndRewrite() local
1114 ++operand_idx; in matchAndRewrite()
1117 exceptions[arg] = operand_idx++; in matchAndRewrite()
1206 auto args = [&](int operand_idx) -> SmallVector<Value, 8> { in GetHoistParams() argument
1209 return arg.getDefiningOp()->getOperand(operand_idx); in GetHoistParams()
1216 auto is_all_tensors = [&](int operand_idx, int axis) -> bool { in GetHoistParams() argument
1219 auto operand = arg.getDefiningOp()->getOperand(operand_idx); in GetHoistParams()
1227 auto is_all_scalars = [&](int operand_idx) -> bool { in GetHoistParams() argument
1230 auto operand = arg.getDefiningOp()->getOperand(operand_idx); in GetHoistParams()
|
| /external/rust/crates/syn/tests/common/ |
| D | eq.rs | 365 spanless_eq_enum!(InlineAsmTemplatePiece; String(0) Placeholder(operand_idx modifier span));
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 2013 for (uint64_t operand_idx = 0; operand_idx < num_elements; operand_idx++) { in fold() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 5099 for (int64 operand_idx = 0; operand_idx < context.operand_shapes.size(); in CheckAndEmitHloWithTile021() local
5100 ++operand_idx) { in CheckAndEmitHloWithTile021()
5101 const Shape& operand_shape = context.operand_shapes[operand_idx]; in CheckAndEmitHloWithTile021()
5116 params_012.push_back(operand_idx); in CheckAndEmitHloWithTile021()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1584 const auto should_partition_contracting_dim = [&](int64 operand_idx) { in PartitionBaseCase() argument
1589 if (operand_idx == 0) { in PartitionBaseCase()
|