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 | 156 ValueRange inputs, Value compilation_key, in CreateSendFromHostOp()
179 TypeRange output_types, Value compilation_key, in CreateRecvAtHostOp()
509 Operation* insertion_point, Value compilation_key, in MoveOpsToHost()
592 Operation* insertion_point, Value compilation_key, in MoveOpsToHost()
648 Value compilation_key, Value device_ordinal, in DecomposeControlFlow()
741 Value compilation_key = compilation_key_op.program(); in CreateParallelExecuteForOutsideCompilation() local
|
| D | tpu_dynamic_layout_pass.cc | 149 Value compilation_key, in BuildGetLayout()
190 const int64_t execute_arg_index, Value compilation_key, in HandleReplicatedInputs()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 456 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() local
473 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() local
|