| /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 | 88 ctx->AssignVariable(0, ctx->input_type(1), ctx->Input(1))); in Compile()
102 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
118 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
175 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 | 739 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 | 288 Status AssignVariable(int input_index, DataType type, xla::XlaOp handle);
290 Status AssignVariable(absl::string_view name, DataType type,
|
| D | xla_op_kernel.cc | 712 Status XlaOpKernelContext::AssignVariable(int input_index, DataType type, in AssignVariable() function in tensorflow::XlaOpKernelContext
719 Status XlaOpKernelContext::AssignVariable(absl::string_view name, DataType type, in AssignVariable() function in tensorflow::XlaOpKernelContext
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/ |
| D | hoist_invariant_ops.mlir | 114 …// CHECK-NEXT: "tf.AssignVariable"([[v]]#[[handle_idx]], [[v]]#[[const_idx]]) {device = "/CPU:0"} …
120 …"tf.AssignVariable"(%handle, %const) {device = "/CPU:0"} : (tensor<!tf_type.resource<tensor<i32>>>…
175 …"tf.AssignVariable"(%var1, %x) {device = "/CPU:0"} : (tensor<!tf_type.resource<tensor<i32>>>, tens…
|
| /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 | 343 // tf.TensorScatterAdd, and tf.AssignVariable.
358 // tf.TensorScatterUpdate, and tf.AssignVariable.
|
| /external/tensorflow/ |
| D | RELEASE.md | 254 * Supports native variable builtin ops - ReadVariable, AssignVariable.
|