Searched defs:shape_i (Results 1 – 6 of 6) sorted by relevance
54 std::vector<size_t> shape_i = AnfAlgo::GetInputDeviceShape(anf_node, i); in Init() local64 std::vector<size_t> shape_i = AnfAlgo::GetOutputDeviceShape(anf_node, i); in Init() local
47 std::string shape_i = "in_shape_" + std::to_string(i); in CodeSessionCompileGraph() local56 std::string shape_i = "out_shape_" + std::to_string(i); in CodeSessionCompileGraph() local
38 std::vector<size_t> shape_i = AnfAlgo::GetInputDeviceShape(anf_node, i); in GetKernelInputShape() local50 std::vector<size_t> shape_i = AnfAlgo::GetOutputDeviceShape(anf_node, i); in GetKernelOutputShape() local
95 std::vector<size_t> shape_i = AnfAlgo::GetInputDeviceShape(anf_node, 0); in GetInputOutputTotalCount() local
47 std::vector<size_t> shape_i = AnfAlgo::GetInputDeviceShape(anf_node, i); in SetIOIputSize() local93 std::vector<size_t> shape_i = AnfAlgo::GetOutputDeviceShape(anf_node, i); in SetIOSize() local
262 std::vector<int64_t> shape_i{1, kUnidirectionalGateNum * hidden_size, input_size}; in SplitWeights() local