Searched refs:input_layout (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | gatherd_info.cc | 119 TensorLayout input_layout; in InferTensorInfo() local
120 …if (input_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[i], inputs_shape_[i]) != SUC… in InferTensorInfo()
124 TensorInfo input_tensor_info(input_layout); in InferTensorInfo()
|
| D | reshape_info.h | 47 void SetInputLayout(const TensorLayout &input_layout) { in SetInputLayout() argument
48 input_layout_ = input_layout; in SetInputLayout()
|
| D | operator_info.cc | 209 TensorLayout input_layout; in InferTensorInfo() local
210 …if (input_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[i], inputs_shape_[i]) != SUC… in InferTensorInfo()
214 TensorInfo input_tensor_info(input_layout); in InferTensorInfo()
|
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | operator_costmodel_test.cc | 167 TensorLayout input_layout, param_layout, output_layout; in TEST_F() local
174 TensorInfo input_info(input_layout, input_shape, input_slice_shape); in TEST_F()
|
| /third_party/gstreamer/gstplugins_bad/tests/examples/d3d11/ |
| D | d3d11device.cpp | 313 ComPtr<ID3D11InputLayout> input_layout; in prepare_shader() local
315 code->GetBufferPointer(), code->GetBufferSize(), &input_layout); in prepare_shader()
397 *layout = input_layout.Detach(); in prepare_shader()
|