Searched refs:compilation_key (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_extract_outside_compilation.cc | 93 ValueRange inputs, Value compilation_key, in CreateSendFromHostOp() argument
99 /*dynamic_key=*/compilation_key, device_ordinal, in CreateSendFromHostOp()
104 /*dynamic_key=*/compilation_key, builder.getStringAttr(communication_key), in CreateSendFromHostOp()
111 TypeRange output_types, Value compilation_key, in CreateRecvAtHostOp() argument
116 loc, output_types, /*dynamic_key=*/compilation_key, device_ordinal, in CreateRecvAtHostOp()
120 loc, output_types, /*dynamic_key=*/compilation_key, in CreateRecvAtHostOp()
266 Operation* insertion_point, Value compilation_key, in MoveOpsToHost() argument
302 builder, op.getLoc(), host_operand_types, compilation_key, in MoveOpsToHost()
309 compilation_key, device_ordinal, in MoveOpsToHost()
358 Value compilation_key, Value device_ordinal, in DecomposeControlFlow() argument
[all …]
|
| D | tpu_dynamic_layout_pass.cc | 142 Value compilation_key, in BuildGetLayout() argument
149 llvm::ArrayRef<Value>{compilation_key}, in BuildGetLayout()
183 const int64_t execute_arg_index, Value compilation_key, in HandleReplicatedInputs() argument
198 auto get_layout = BuildGetLayout(execute_arg_index, compilation_key, in HandleReplicatedInputs()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 428 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() local
445 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() local
446 compilation_key.flat<tstring>()(0) = key; in Compute()
451 ctx->set_output(0, compilation_key); in Compute()
|