Lines Matching defs:arg
80 for (const auto& arg : signature.arg_types) {
81 h = Hash64Combine(h, std::hash<int>()(static_cast<int>(arg.first)));
82 h = Hash64Combine(h, std::hash<int>()(arg.second.dims()));
83 for (int dim : arg.second.dim_sizes()) {
87 for (const auto& arg : signature.arg_values) {
89 h, Hash64(arg.tensor_data().data(), arg.tensor_data().size()));
148 XlaCompiler::Argument& arg = (*args)[input_num];
149 arg.kind = XlaCompiler::Argument::kConstant;
150 arg.type = input.dtype();
151 arg.shape = input.shape();
152 arg.constant_value = input;
164 XlaCompiler::Argument& arg = (*args)[input_num];
166 arg.kind = XlaCompiler::Argument::kParameter;
168 arg.kind = XlaCompiler::Argument::kConstant;
169 arg.constant_value = input;
171 arg.type = input.dtype();
172 arg.shape = input.shape();
182 XlaCompiler::Argument& arg = (*args)[input_num];
184 arg.name = variable_args[variable_id].name;
185 arg.kind = XlaCompiler::Argument::kResource;
186 arg.resource_kind = XlaResource::kVariable;
189 arg.type = value.dtype();
190 arg.shape = value.shape();
191 arg.initialized = true;
197 arg.initialized = false;
198 arg.type = DT_INVALID;
199 arg.shape = TensorShape();
