Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dtensor_to_scalar.h30TensorToScalar() : BaseOperator(kNameTensorToScalar) { InitIOName({"input_tensor"}, {"output_data"… in TensorToScalar() function
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/python_input/gtest_input/pipeline/infer/
Dprimitive_test.py43 class TensorToScalar(PrimitiveWithInfer): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/
Dbprop_irbuilder.cc250 NodePtr BpropBuilder::TensorToScalar(const NodePtr &node) { in TensorToScalar() function in mindspore::expander::bprop::BpropBuilder
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
D_sequence_ops.py485 class TensorToScalar(PrimitiveWithCheck): class