Home
last modified time | relevance | path

Searched defs:input_map (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dmindrt_executor.h35 std::unordered_map<Tensor *, Tensor *> *input_map) in MindrtExecutor()
Dlite_mindrt.h61 void set_isolate_input_map(std::unordered_map<Tensor *, Tensor *> *input_map) { in set_isolate_input_map() argument
Dlite_mindrt.cc99 std::unordered_map<Tensor *, Tensor *> *input_map) { in PreInit() argument
108 std::unordered_map<Tensor *, Tensor *> *input_map) { in IsolateInputData() argument
/third_party/mesa3d/src/gallium/drivers/svga/
Dsvga_link.h21 uint8_t input_map[PIPE_MAX_SHADER_INPUTS]; member
Dsvga_tgsi_emit.h90 struct src_register input_map[PIPE_MAX_ATTRIBS]; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcast.cc43 Eigen::TensorMap<Eigen::Tensor<T, kRank, Eigen::RowMajor>> input_map(input_addr, 1, input_size); in CastTask() local
Dmeshgrid.cc51 …Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor>> input_map(reinterpret_cast<T *>(ctx.Input(i… in MeshgridTask() local
Drandom_categorical.cc40 …Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor>> input_map(static_cast<T *>(logits_data), ba… in RandomCategoricalCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dgptq_quantizer.h90 std::set<TK> extract_keys(std::map<TK, TV> const &input_map) { in extract_keys()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dge_adapter_info.cc179 auto input_map = adapter_->getInputMap(); in InitAttrToInputMap() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/actor/
Dexit_actor.cc127 std::unordered_map<Tensor *, Tensor *> *input_map) { in PreInit() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc174 mindspore::HashMap<int, std::string> input_map; in GenerateCustomOpInputMap() local
318 mindspore::HashMap<int, std::string> &input_map = it->second; in SetCustomOpInput() local
355 mindspore::HashMap<int, std::string> &input_map = it->second; in SetCustomOpInput() local
732 mindspore::HashMap<int, std::string> &input_map = (*cus_input_map_)[op->GetOpType()]; in UpdateCustomOpInputDesc() local
Dutils.cc514 auto input_map = adpt->getInputMap(); in SinkGraphCheck() local
Dconvert.cc2917 const auto &input_map = adpt->getInputMap(); in IsDynamicInputBeforeNormalInput() local
3177 const auto &input_map = adpt->getInputMap(); in SetOpAttrToInput() local
4460 auto input_map = adpt->getInputMap(); in DrawCNode() local
Dop_adapter.h38 OpAdapterImpl(const mindspore::HashMap<int, InputDesc> &input_map, in OpAdapterImpl()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dinputs_unify_mindir.cc80 auto input_map = adpt->getInputMap(); in Process() local
/third_party/mesa3d/src/gallium/frontends/nine/
Dvertexshader9.h26 } input_map[PIPE_MAX_ATTRIBS]; member
Dnine_shader.h41 uint16_t input_map[PIPE_MAX_ATTRIBS]; /* VS input -> NINE_DECLUSAGE_x */ member
/third_party/mesa3d/src/compiler/nir/
Dnir_lower_io_to_temporaries.c46 struct hash_table *input_map; member
/third_party/ffmpeg/libavresample/
Dinternal.h50 int input_map[AVRESAMPLE_MAX_CHANNELS]; /**< dest index of each input channel */ member
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dconfig_file_parser.cc605 int ConfigFileParser::SetMapData(const std::map<std::string, std::string> &input_map, in SetMapData()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_assigned_parallel.cc782 mindspore::HashMap<AnfNodePtr, int> input_map = {}; in StepReplaceGraph() local
Dstep_parallel.cc908 mindspore::HashMap<AnfNodePtr, int> input_map = {}; in StepReplaceGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/kernels/
Daudio_utils.cc1161 auto input_map = Map<MatrixXT>(input_data, num_wavs, wave_length); in Pad() local
1242 …auto input_map = Map<MatrixXT, 0, Eigen::OuterStride<1>>(input_data + freq * (n_frame + 2 * n), n_… in ComputeDeltasImpl() local