HomeSort by relevance Sort by last modified time
    Searched refs:AssignVariable (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 52 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle));
109 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var));
110 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum));
156 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var));
157 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum));
242 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, var));
243 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, dtype_, m));
244 OP_REQUIRES_OK(ctx, ctx->AssignVariable(2, dtype_, v));
334 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, new_var));
335 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, new_ms))
    [all...]
variable_ops.cc 67 ctx->AssignVariable(0, ctx->input_type(1), ctx->Input(1)));
81 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle));
97 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle));
strided_slice_op.cc 321 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, lhs));
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_kernel.h 177 Status AssignVariable(int input_index, DataType type,
xla_op_kernel.cc 412 Status XlaOpKernelContext::AssignVariable(int input_index, DataType type,

Completed in 245 milliseconds