Searched refs:packed_inputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 218 llvm::SmallVectorImpl<OpAsmParser::UnresolvedOperand>* packed_inputs, in ParseReplicateOpOperands() argument
240 packed_inputs->emplace_back(operand_type); in ParseReplicateOpOperands()
280 llvm::ArrayRef<OpAsmParser::UnresolvedOperand> packed_inputs, in SetReplicateOpOperands() argument
290 if (replicated_inputs.empty() && packed_inputs.empty()) return success(); in SetReplicateOpOperands()
308 for (auto packed_input_and_idx : llvm::enumerate(packed_inputs)) { in SetReplicateOpOperands()
332 llvm::SmallVector<OpAsmParser::UnresolvedOperand, 8> packed_inputs; in parse() local
338 &packed_inputs, ®ion_args, in parse()
342 packed_inputs, region_arg_types, &n)) in parse()
356 int32_t num_packed_inputs = packed_inputs.size(); in parse()
404 p.printOperand(*std::next(packed_inputs().begin(), in print()
[all …]
|
| D | tf_device_ops.td | 209 packed_inputs: each input corresponds to an input broadcasted across all
271 Variadic<AnyType>:$packed_inputs,
302 "ValueRange":$packed_inputs, "TypeRange":$replica_output_types)>,
305 "ValueRange":$packed_inputs, "TypeRange":$replica_output_types)>,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 423 llvm::SmallVector<Value, 8> packed_inputs; in ReplicateCluster() local
434 packed_inputs.push_back(input->getOperand(0)); in ReplicateCluster()
465 replicated_inputs, packed_inputs, cluster.getResultTypes()); in ReplicateCluster()
|