HomeSort by relevance Sort by last modified time
    Searched full:op_kernel_context (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
arg_op.cc 47 ctx->op_kernel_context()->set_output(0, val);
retval_op.cc 36 const Tensor& input = ctx->op_kernel_context()->input(0);
while_op.cc 295 ctx->op_kernel_context()->set_output(
297 ctx->op_kernel_context()->input(update.input_index));
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_context.cc 58 return Get(ctx->op_kernel_context());
xla_op_kernel.h 197 OpKernelContext* op_kernel_context() const { return context_; } function in class:tensorflow::XlaOpKernelContext
xla_compiler_test.cc 91 ResourceMgr* rm = ctx->op_kernel_context()->resource_manager();

Completed in 3047 milliseconds