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