Home
last modified time | relevance | path

Searched defs:temp_shape (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/
Dlookup_ops_shape_fns.cc65 Shape temp_shape = output_shape_and_type.GetShape(); in ValidateTableResourceHandle() local
74 Shape temp_shape = output_shape_and_type.GetShape(); in ValidateTableResourceHandle() local
154 auto temp_shape = output_shape_and_type.GetShape(); in ValidateTableResourceHandle() local
165 auto temp_shape = output_shape_and_type.GetShape(); in ValidateTableResourceHandle() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsequence_unstack.cc69 auto temp_shape = x_shape; in SequenceUnstackInferInner() local
84 auto temp_shape = x_shape; in SequenceUnstackInferInner() local
Dunstack.cc65 auto temp_shape = x_shape; in UnstackInferShape() local
99 auto temp_shape = x_shape; in UnstackInferInner() local
Dnuclear_norm.cc74 auto temp_shape = input_shape; in NuclearNormInferShape() local
Dbuffer_append.cc117 std::vector<int64_t> temp_shape(d_shape.begin() + 1, d_shape.end()); in BufferAppendInferShape() local
Dget_tuple_index_info.cc150 ShapeVector temp_shape(broadcast_shape_len, 1); in ComputeSliceShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dshuffle_channel.cc118 std::vector<int64_t> temp_shape; in ShuffleChannelCompute() local
Dpack.cc110 std::vector<int64_t> temp_shape = input->GetTensorShape()->GetDimSizes(); in PackCompute() local
Dsparse_addmm.cc286 std::vector<int64_t> temp_shape; in SparseAddmmCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_concat_cpu_kernel.cc206 …const auto &temp_shape = reinterpret_cast<T *>(inputs[kSpInputShapesStart * size + i]->device_ptr(… in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_kernel_runtime.cc202 ShapeVector temp_shape; in CreateTensorForOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc265 ShapeVector temp_shape; in CreateTLoopNodeWithEdge() local
/third_party/skia/third_party/externals/freetype/src/sdf/
Dftsdf.c3507 SDF_Shape temp_shape; /* temporary shape */ in sdf_generate_with_overlaps() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dms_device_shape_transfer.cc504 auto temp_shape = shape; in TransCore() local