Searched refs:GetVariableTypeAndShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | variable_ops.cc | 38 ctx->GetVariableTypeAndShape(0, &variable_dtype, &variable_shape)); in ValidateAssignUpdateVariableOpShapes()
66 ctx->GetVariableTypeAndShape(0, &variable_dtype, &shape)); in Compile()
|
| D | sharding_util_ops.cc | 280 ctx, ctx->GetVariableTypeAndShape(/*index=*/0, &variable_input_dtype, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 262 Status GetVariableTypeAndShape(int index, DataType* type,
|
| D | xla_op_kernel.cc | 605 Status XlaOpKernelContext::GetVariableTypeAndShape(int index, DataType* type, in GetVariableTypeAndShape() function in tensorflow::XlaOpKernelContext
|