Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_builder.cc341 auto tensor_dtype = abs->BuildType()->cast<TensorTypePtr>(); in FillZeros() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecial_op_eliminate.h640 auto tensor_dtype = constant_tensor->Dtype(); in operator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/
Ddebug_services.cc161 uint64_t prev_num_elements, int tensor_dtype) { in GetSummaryPtr()
583 int tensor_dtype = tensor->GetType(); in CheckWatchpointsForTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc770 TypePtr tensor_dtype = arg->element()->BuildType(); in InferImplMakeSlice() local