Searched defs: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()
111 TypeRange output_types, Value compilation_key, in CreateRecvAtHostOp()
266 Operation* insertion_point, Value compilation_key, in MoveOpsToHost()
358 Value compilation_key, Value device_ordinal, in DecomposeControlFlow()
441 Value compilation_key = compilation_key_op.program(); in CreateParallelExecuteForOutsideCompilation() local
|
| D | tpu_dynamic_layout_pass.cc | 142 Value compilation_key, in BuildGetLayout()
183 const int64_t execute_arg_index, Value 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
|