| /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 | 87 ctx, ctx->ReadVariableInput(0, dtype_, /*shape=*/nullptr, &handle)); in Compile()
113 ctx->ReadVariableInput(0, type, /*shape=*/nullptr, &handle)); in Compile()
130 ctx->ReadVariableInput(0, type, /*shape=*/nullptr, &handle)); in Compile()
150 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &input_shape, &input)); in Compile()
181 context, context->ReadVariableInput(0, dtype, &var_shape, &var_value)); in Compile()
|
| D | stateful_random_ops.cc | 192 TF_RETURN_IF_ERROR(ctx->ReadVariableInput( in CompileImpl()
440 ctx->ReadVariableInput(state_input_idx, STATE_ELEMENT_DTYPE, in Compile()
|
| D | sharding_util_ops.cc | 289 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(/*index=*/0, dtype_, in Compile()
|
| D | strided_slice_op.cc | 677 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &lhs_shape, &lhs)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 282 Status ReadVariableInput(int index, DataType type, TensorShape* shape,
286 Status ReadVariableInput(absl::string_view name, DataType type,
|
| D | xla_op_kernel.cc | 591 Status XlaOpKernelContext::ReadVariableInput(int index, DataType type, in ReadVariableInput() function in tensorflow::XlaOpKernelContext
598 Status XlaOpKernelContext::ReadVariableInput(absl::string_view name, in ReadVariableInput() function in tensorflow::XlaOpKernelContext
|