Searched refs:XLAShapeForArgument (Results 1 – 2 of 2) sorted by relevance
359 Status XLAShapeForArgument(const Argument& arg, bool is_entry_computation,
612 Status XlaCompiler::XLAShapeForArgument(const XlaCompiler::Argument& arg, in XLAShapeForArgument() function in tensorflow::XlaCompiler770 TF_RETURN_IF_ERROR(XLAShapeForArgument( in BuildArguments()