Searched defs:node_context (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| H A D | conv_pointwise.cc | 82 NodeContext* node_context) { in IsNode()
111 NodeContext* node_context) { in IsMeanNode()
122 NodeContext* node_context) { in IsMulNode()
132 NodeContext* node_context) { in IsSliceNode()
144 NodeContext* node_context) { in IsConcatNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_execute_op.cc | 138 tpu::TpuNodeContext* node_context, se::Stream* stream, in AllocateOutputTuple()
173 tpu::TpuNodeContext* node_context, se::Stream* stream, in AllocateOutputTensors()
189 OpKernelContext* context, tpu::TpuNodeContext* node_context, in RunExecutable()
231 OpKernelContext* context, tpu::TpuNodeContext* node_context, in ExecuteTPUProgram()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | shape_refiner.cc | 71 InferenceContext* node_context = CHECK_NOTNULL(GetContext(node)); in InferShapesForFunctionSubNode() local
313 InferenceContext* node_context = node_ext_context->get_context(); in UpdateNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_node_context.h | 48 explicit TpuNodeContext(int device_ordinal, XLA_TpuNodeContext* node_context) in TpuNodeContext()
|
| H A D | tpu_node_context.cc | 30 XLA_TpuNodeContext* node_context = in Create() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 422 TpuNodeContext* node_context, xla::DeviceAssignment* device_assignment, in TPUExecute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_execute_op.cc | 396 const VariableUpdateMap& variable_updates, TpuNodeContext* node_context, in AllocateOutputTensors()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 913 auto* node_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner() local
928 auto* node_context = shape_refiner_->GetContext(node); in AddNodesToShapeRefiner() local
|