Searched refs:XLAShapeForArgument (Results 1 – 2 of 2) sorted by relevance
290 Status XLAShapeForArgument(const Argument& arg, xla::Shape* xla_shape);
232 Status XlaCompiler::XLAShapeForArgument(const XlaCompiler::Argument& arg, in XLAShapeForArgument() function in tensorflow::XlaCompiler488 XLAShapeForArgument(args[(*input_mapping)[i]], &arg_shapes[i])); in BuildArguments()