Searched refs:AssignVariable (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 52 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
107 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, var)); in Compile()
164 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var)); in Compile()
165 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum)); in Compile()
222 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var)); in Compile()
223 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum)); in Compile()
266 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var)); in Compile()
267 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum)); in Compile()
318 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, var)); in Compile()
319 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, dtype_, accum)); in Compile()
[all …]
|
| D | variable_ops.cc | 86 ctx->AssignVariable(0, ctx->input_type(1), ctx->Input(1))); in Compile()
100 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
116 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
176 context->AssignVariable(0, dtype, result.ValueOrDie())); in Compile()
|
| D | strided_slice_op.cc | 318 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, lhs)); in Compile()
|
| D | stateful_random_ops.cc | 239 ctx->AssignVariable(state_input_idx, STATE_ELEMENT_DTYPE, var)); in CompileImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 215 Status AssignVariable(int input_index, DataType type, xla::XlaOp handle);
217 Status AssignVariable(absl::string_view name, DataType type,
|
| D | xla_op_kernel.cc | 543 Status XlaOpKernelContext::AssignVariable(int input_index, DataType type, in AssignVariable() function in tensorflow::XlaOpKernelContext
550 Status XlaOpKernelContext::AssignVariable(absl::string_view name, DataType type, in AssignVariable() function in tensorflow::XlaOpKernelContext
|