Searched defs:node_context (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_node_context.h | 50 explicit TpuNodeContext(int device_ordinal, XLA_TpuNodeContext* node_context) in TpuNodeContext()
|
| D | tpu_node_context.cc | 32 XLA_TpuNodeContext* node_context = in Create() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 138 tpu::TpuNodeContext* node_context, se::Stream* stream, in AllocateOutputTuple()
172 tpu::TpuNodeContext* node_context, se::Stream* stream, in AllocateOutputTensors()
188 OpKernelContext* context, tpu::TpuNodeContext* node_context, in RunExecutable()
230 OpKernelContext* context, tpu::TpuNodeContext* node_context, in ExecuteTPUProgram()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 71 InferenceContext* node_context = CHECK_NOTNULL(GetContext(node)); in InferShapesForFunctionSubNode() local
297 InferenceContext* node_context = node_ext_context->get_context(); in UpdateNode() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 418 TpuNodeContext* node_context, xla::DeviceAssignment* device_assignment, in TPUExecute()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 397 const VariableUpdateMap& variable_updates, TpuNodeContext* node_context, in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 792 auto* node_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner() local
827 auto* node_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner() local
|