Home
last modified time | relevance | path

Searched defs:zero_tensor (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dshuffle_tensorrt.cc339 auto zero_tensor = ctx_->ConvertTo1DTensor(0); in AddBroadcastToOp() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc536 …auto zero_tensor = TensorConstructUtils::CreateZerosTensor(out_shape_type.second, out_shapes.at(in… in CreateZeroseOutput() local
1002 … auto zero_tensor = TensorConstructUtils::CreateZerosTensor(attrs[DTYPE]->cast<TypePtr>(), {1}); in GetSendsRecvs() local
Dpipeline_transformer.cc1337 auto zero_tensor = TensorConstructUtils::CreateZerosTensor(out_shape_type.second, out_shape); in CreateZeroseOutput() local