Searched refs:input_layout_ (Results 1 – 2 of 2) sorted by relevance
51 Shape input_tensor_map = input_layout_.tensor_map().array(); in InferMirrorOps()118 if (tensor_redistribution.Init(input_layout_, output_layout_, dev_list) == FAILED) { in ComputeReplaceOp()126 MS_LOG(DEBUG) << name_ << ": input " << input_layout_.ToString(); in ComputeReplaceOp()236 input_layout_ = tensor_layout_in; in InferTensorLayout()280 TensorInfo tensor_info_in(input_layout_); in InferTensorInfoByLayout()330 Status status = InferDefaultLayout(inputs_shape_.at(0), &input_layout_); in Init()344 inputs_tensor_map_.push_back(input_layout_.tensor_map().array()); in Init()348 dev_matrix_shape_ = input_layout_.device_arrangement().array(); in Init()
48 input_layout_ = input_layout; in SetInputLayout()97 TensorLayout input_layout_; variable