Home
last modified time | relevance | path

Searched refs:ReadVariableInput (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc37 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &handle)); in Compile()
82 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &var_shape, &var)); in Compile()
128 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
129 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(1, type, &accum_shape, &accum)); in Compile()
186 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
187 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(1, type, &accum_shape, &accum)); in Compile()
242 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
243 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(1, type, &accum_shape, &accum)); in Compile()
283 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &var_shape, &var)); in Compile()
285 ctx->ReadVariableInput(1, dtype_, &accum_shape, &accum)); in Compile()
[all …]
Dvariable_ops.cc72 ctx, ctx->ReadVariableInput(0, dtype_, /*shape=*/nullptr, &handle)); in Compile()
98 ctx->ReadVariableInput(0, type, /*shape=*/nullptr, &handle)); in Compile()
114 ctx->ReadVariableInput(0, type, /*shape=*/nullptr, &handle)); in Compile()
133 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &resource_shape, in Compile()
167 context, context->ReadVariableInput(0, dtype, &var_shape, &var_value)); in Compile()
Dstrided_slice_op.cc260 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &lhs_shape, &lhs)); in Compile()
Dstateful_random_ops.cc199 TF_RETURN_IF_ERROR(ctx->ReadVariableInput( in CompileImpl()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h204 Status ReadVariableInput(int index, DataType type, TensorShape* shape,
208 Status ReadVariableInput(absl::string_view name, DataType type,
Dxla_op_kernel.cc406 Status XlaOpKernelContext::ReadVariableInput(int index, DataType type, in ReadVariableInput() function in tensorflow::XlaOpKernelContext
413 Status XlaOpKernelContext::ReadVariableInput(absl::string_view name, in ReadVariableInput() function in tensorflow::XlaOpKernelContext