Home
last modified time | relevance | path

Searched defs:input_b_shape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Darithmetic_info.cc41 Shape input_b_shape = inputs_shape_.at(1); in InferExpandShape() local
90 Shape input_b_shape = input_shapes.at(1); in BaseCheckStrategy() local
403 Shape input_b_shape = input_shapes.at(1); in InferOutputLayout() local
Dgrid_sampler2d.cc63 Shape input_b_shape = inputs_shape_.at(1); in CheckStrategy() local
Dunsorted_segment_op_info.cc80 Shape input_b_shape = inputs_shape_.at(1); in CheckStrategy() local
Dreduce_method_info.cc544 Shape input_b_shape = inputs_shape_.at(1); in CheckStrategy() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc843 Shape input_b_shape = input_b_tensor_info.shape(); in GetBackwardComputationCost() local
880 Shape input_b_shape = input_b_tensor_info.shape(); in GetBackwardCommCost() local