| /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()
270 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var)); in Compile()
271 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum)); in Compile()
325 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var)); in Compile()
326 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum)); in Compile()
[all …]
|
| D | variable_ops.cc | 87 ctx->AssignVariable(0, ctx->input_type(1), ctx->Input(1))); in Compile()
101 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
117 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
174 context->AssignVariable(0, dtype, result.ValueOrDie())); in Compile()
|
| D | stateful_random_ops.cc | 212 ctx->AssignVariable(state_input_idx, STATE_ELEMENT_DTYPE, var)); in CompileImpl()
471 ctx, ctx->AssignVariable(state_input_idx, STATE_ELEMENT_DTYPE, var)); in Compile()
|
| D | strided_slice_op.cc | 701 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, lhs)); in Compile()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.h | 42 Status AssignVariable(ImmediateExecutionContext* ctx,
|
| D | variable_ops_test.cc | 84 TF_EXPECT_OK(internal::AssignVariable(context(), variable.get(), DT_FLOAT, in TEST_F()
|
| D | variable_ops.cc | 71 Status AssignVariable(ImmediateExecutionContext* ctx, in AssignVariable() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 243 Status AssignVariable(int input_index, DataType type, xla::XlaOp handle);
245 Status AssignVariable(absl::string_view name, DataType type,
|
| D | xla_op_kernel.cc | 652 Status XlaOpKernelContext::AssignVariable(int input_index, DataType type, in AssignVariable() function in tensorflow::XlaOpKernelContext
659 Status XlaOpKernelContext::AssignVariable(absl::string_view name, DataType type, in AssignVariable() function in tensorflow::XlaOpKernelContext
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | variable.cc | 60 return internal::AssignVariable(ctx_, handle_.get(), dtype_, handle); in Assign()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_colocate_composite_resource_ops.mlir | 66 // Tests AssignVariable op using composite device resource is wrapped inside
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.td | 330 // tf.TensorScatterUpdate, and tf.AssignVariable.
|