Searched refs:BoolImmPtr (Results 1 – 16 of 16) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | cc_implementations_test.cc | 277 ASSERT_EQ(ScalarEq(list)->cast<BoolImmPtr>()->value(), true); in TEST_F()
282 ASSERT_EQ(ScalarEq(list)->cast<BoolImmPtr>()->value(), false); in TEST_F()
287 ASSERT_EQ(ScalarEq(list)->cast<BoolImmPtr>()->value(), true); in TEST_F()
292 ASSERT_EQ(ScalarEq(list)->cast<BoolImmPtr>()->value(), true); in TEST_F()
300 ASSERT_EQ(ScalarLt(list)->cast<BoolImmPtr>()->value(), false); in TEST_F()
305 ASSERT_EQ(ScalarLt(list)->cast<BoolImmPtr>()->value(), false); in TEST_F()
310 ASSERT_EQ(ScalarLt(list)->cast<BoolImmPtr>()->value(), true); in TEST_F()
315 ASSERT_EQ(ScalarLt(list)->cast<BoolImmPtr>()->value(), true); in TEST_F()
323 ASSERT_EQ(ScalarGt(list)->cast<BoolImmPtr>()->value(), false); in TEST_F()
328 ASSERT_EQ(ScalarGt(list)->cast<BoolImmPtr>()->value(), true); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | matmul_info.cc | 80 transpose_a_ = transpose_a_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()
91 transpose_b_ = transpose_b_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()
102 forward_reduce_scatter_ = forward_reduce_scatter_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()
|
| D | uniform_candidate_sampler_info.cc | 59 *value = iter->second->cast<BoolImmPtr>()->value(); in GetUniformSamplerAttrBool()
|
| D | reduce_method_info.cc | 89 keepdims_ = keep_dims_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()
99 cross_batch_ = cross_batch_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()
|
| D | reshape_info.cc | 317 is_skip_ = reshape_skip_redis_iter->second->cast<BoolImmPtr>()->value(); in Init()
|
| D | operator_info.cc | 1792 return attr_iter->second->cast<BoolImmPtr>()->value(); in GetBoolAttr()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | scalar.h | 81 using BoolImmPtr = std::shared_ptr<BoolImm>; variable
82 IMM_TRAITS(BoolImmPtr, bool) in IMM_TRAITS() argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_parse_graph.cc | 123 bool transpose_a = attrs[TRANSPOSE_A]->cast<BoolImmPtr>()->value(); in Complete2DInputs()
124 bool transpose_b = attrs[TRANSPOSE_B]->cast<BoolImmPtr>()->value(); in Complete2DInputs()
|
| D | rec_generate_strategy.cc | 69 bool transpose_a = attrs[TRANSPOSE_A]->cast<BoolImmPtr>()->value(); in PrepareMatMul()
70 bool transpose_b = attrs[TRANSPOSE_B]->cast<BoolImmPtr>()->value(); in PrepareMatMul()
814 keepdims = keep_dims_iter->second->cast<BoolImmPtr>()->value(); in GetKeepDims()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | convert_utils.cc | 35 *value = v->cast<BoolImmPtr>()->value(); in ValueToBool()
|
| D | convert_utils_py.cc | 103 bool_v = value->cast<BoolImmPtr>()->value(); in ScalarPtrToPyData()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_structures.cc | 119 …ValuePtr scalar_index = MakeValue(static_cast<int64_t>(scalar_value->cast<BoolImmPtr>()->value())); in InferImplMakeSlice()
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | fetch_content.cc | 174 auto data = value->cast<mindspore::BoolImmPtr>(); in FetchFromBoolImmValue()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | proto_exporter.cc | 180 const BoolImmPtr &value = dyn_cast<BoolImm>(val); in SetScalarToProto()
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | dump_proto.cc | 245 const BoolImmPtr &value = dyn_cast<BoolImm>(val); in SetScalarToProto()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 1273 bool data_parallel = data_parallel_iter->second->cast<BoolImmPtr>()->value(); in OperatorInstanceByName()
|