Searched defs:param_shape (Results 1 – 8 of 8) sorted by relevance
34 param_shape=(8, 8)): argument
41 param_shape=(8, 8)): argument
136 auto param_shape = AnfAlgo::GetPrevNodeOutputInferShape(origin_node, 0); in CheckInputs() local
169 Shape param_shape = {32}; in TEST_F() local
321 auto param_shape = inputs_shape_.at(0); in CheckStrategy() local
920 std::vector<size_t> param_shape = input_shape; in MS_TRT_CONVERTER_FUNC_REG() local
1507 auto param_shape = inputs[0].slice_shape(); in GetForwardCommCost() local
1256 const auto ¶m_shape = AnfAlgo::GetOutputInferShape(param, 0); in GetParameterIndex() local