Searched refs:inputs_mask (Results 1 – 2 of 2) sorted by relevance
65 std::vector<int64_t> inputs_mask; member
525 if (input_num != op_run_info->inputs_mask.size()) { in ConstructInputTensor()527 << op_run_info->inputs_mask.size(); in ConstructInputTensor()536 int64_t tensor_mask = op_run_info->inputs_mask[index]; in ConstructInputTensor()539 op_run_info->inputs_mask[index] = tensor_mask; in ConstructInputTensor()1030 op_exec_info->inputs_mask = std::move(op_masks); in ConstructForwardGraph()1540 for (const auto mask : op_exec_info->inputs_mask) { in RecordGradOpInfo()