Home
last modified time | relevance | path

Searched defs:tensor_dtype (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprim_structures.cc127 TypePtr tensor_dtype = arg->element()->BuildType(); in InferImplMakeSlice() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc269 std::string tensor_dtype = "float16"; in GenTopKV2IndicesTensorInfo() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecial_op_eliminate.h669 auto tensor_dtype = constant_tensor->Dtype(); in operator() local
/third_party/mindspore/mindspore/ccsrc/debug/
Ddebug_services.cc89 uint32_t prev_num_elements, int tensor_dtype) { in GetSummaryPtr()
369 int tensor_dtype = tensor->GetType(); in CheckWatchpointsForTensor() local