Searched refs:argument_shapes_size (Results 1 – 2 of 2) sorted by relevance
52 const int64 argument_shapes_size = argument_shapes.size(); in CreateModuleConfig() local53 if (program_shape.parameters_size() != argument_shapes_size) { in CreateModuleConfig()
125 const int argument_shapes_size = argument_shapes.size(); in RunHelper() local126 if (argument_shapes_size != computation_layout.parameter_count()) { in RunHelper()