Searched refs:is_weight (Results 1 – 7 of 7) sorted by relevance
23 BOOST_TEST_NOT(dtl::is_weight<A>::value); in main()24 BOOST_TEST_NOT(dtl::is_weight<B>::value); in main()25 BOOST_TEST_NOT(dtl::is_weight<decltype(sample(0))>::value); in main()26 BOOST_TEST(dtl::is_weight<decltype(weight(0))>::value); in main()
27 using is_weight = is_weight_impl<T>; typedef62 static constexpr std::size_t weight_ = mp11::mp_find_if<list_, is_weight>::value;
237 bool is_weight = AnfAlgo::IsParameterWeight(parameter); in CreateKernelInfo() local238 kernel_info->set_feature_map_flag(!is_weight); in CreateKernelInfo()239 if (!is_weight) { in CreateKernelInfo()
717 bool is_weight = input_node->cast<ParameterPtr>()->has_default(); in BuildKNodeForCNodeInput() local719 if (is_weight && need_grad_weights_.find(input_node) == need_grad_weights_.end()) { in BuildKNodeForCNodeInput()757 bool is_weight = input_node->cast<ParameterPtr>()->has_default(); in BackPropagateOneCNodeWithBPropFuncGraph() local758 if (!is_weight || need_grad_weights_.find(input_node) != need_grad_weights_.end()) { in BackPropagateOneCNodeWithBPropFuncGraph()798 bool is_weight = input_node->cast<ParameterPtr>()->has_default(); in BackPropagateOneCNodeWithFPropFuncGraph() local799 if (!is_weight || need_grad_weights_.find(input_node) != need_grad_weights_.end()) { in BackPropagateOneCNodeWithFPropFuncGraph()
542 bool is_weight = AnfAlgo::IsParameterWeight(parameter); in SetKernelInfoForNode() local543 kernel_info->set_feature_map_flag(!is_weight); in SetKernelInfoForNode()544 types.push_back(is_weight ? kTypeUnknown : AnfAlgo::GetOutputInferDataType(parameter, 0)); in SetKernelInfoForNode()
82 bool is_weight; member
174 *output_is_weight = iter->second.is_weight; in GetCNodeOutputStubTensor()217 output_tensor_info.is_weight = !(kernel_info->is_feature_map()); in GenOpOutputStubTensor()