Searched refs:GetVariableTypeAndShape (Results 1 – 3 of 3) sorted by relevance
212 Status GetVariableTypeAndShape(int index, DataType* type,
524 Status XlaOpKernelContext::GetVariableTypeAndShape(int index, DataType* type, in GetVariableTypeAndShape() function in tensorflow::XlaOpKernelContext
53 ctx->GetVariableTypeAndShape(0, &variable_dtype, &shape)); in Compile()