Searched refs:XLAShapeForArgument (Results 1 – 3 of 3) sorted by relevance
229 Status XLAShapeForArgument(
833 Status XlaCompiler::XLAShapeForArgument( in XLAShapeForArgument() function in tensorflow::XlaCompiler1033 TF_RETURN_IF_ERROR(XLAShapeForArgument(args[(*input_to_args)[i]], in BuildArguments()
244 TF_RETURN_IF_ERROR(compiler.XLAShapeForArgument( in BuildComputationArgumentDescriptions()