Searched refs:packed_inputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 217 llvm::SmallVectorImpl<OpAsmParser::OperandType>* packed_inputs, in ParseReplicateOpOperands() argument
239 packed_inputs->emplace_back(operand_type); in ParseReplicateOpOperands()
278 llvm::ArrayRef<OpAsmParser::OperandType> packed_inputs, in SetReplicateOpOperands() argument
288 if (replicated_inputs.empty() && packed_inputs.empty()) return success(); in SetReplicateOpOperands()
306 for (auto packed_input_and_idx : llvm::enumerate(packed_inputs)) { in SetReplicateOpOperands()
328 llvm::SmallVector<OpAsmParser::OperandType, 8> packed_inputs; in ParseReplicateOp() local
334 &packed_inputs, ®ion_args, in ParseReplicateOp()
338 packed_inputs, region_arg_types, &n) || in ParseReplicateOp()
345 int32_t num_packed_inputs = packed_inputs.size(); in ParseReplicateOp()
397 p->printOperand(*std::next(op.packed_inputs().begin(), in Print()
[all …]
|
| D | tf_device_ops.td | 207 packed_inputs: each input corresponds to an input broadcasted across all
269 Variadic<AnyType>:$packed_inputs,
300 "ValueRange":$packed_inputs, "TypeRange":$replica_output_types)>,
303 "ValueRange":$packed_inputs, "TypeRange":$replica_output_types)>,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 394 llvm::SmallVector<Value, 8> packed_inputs; in ReplicateCluster() local
406 packed_inputs.push_back(input->getOperand(0)); in ReplicateCluster()
427 replicated_inputs, packed_inputs, cluster.getResultTypes()); in ReplicateCluster()
|