| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 37 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()
244 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
245 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(1, type, &accum_shape, &accum)); in Compile()
293 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
294 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(1, type, &accum_shape, &accum)); in Compile()
[all …]
|
| D | variable_ops.cc | 73 ctx, ctx->ReadVariableInput(0, dtype_, /*shape=*/nullptr, &handle)); in Compile()
99 ctx->ReadVariableInput(0, type, /*shape=*/nullptr, &handle)); in Compile()
115 ctx->ReadVariableInput(0, type, /*shape=*/nullptr, &handle)); in Compile()
134 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &input_shape, &input)); in Compile()
165 context, context->ReadVariableInput(0, dtype, &var_shape, &var_value)); in Compile()
|
| D | stateful_random_ops.cc | 191 TF_RETURN_IF_ERROR(ctx->ReadVariableInput( in CompileImpl()
439 ctx->ReadVariableInput(state_input_idx, STATE_ELEMENT_DTYPE, in Compile()
|
| D | strided_slice_op.cc | 638 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &lhs_shape, &lhs)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 232 Status ReadVariableInput(int index, DataType type, TensorShape* shape,
236 Status ReadVariableInput(absl::string_view name, DataType type,
|
| D | xla_op_kernel.cc | 510 Status XlaOpKernelContext::ReadVariableInput(int index, DataType type, in ReadVariableInput() function in tensorflow::XlaOpKernelContext
517 Status XlaOpKernelContext::ReadVariableInput(absl::string_view name, in ReadVariableInput() function in tensorflow::XlaOpKernelContext
|