Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_to_dense_cpu_kernal.cc38 auto values_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
Dsparse_tensor_dense_matmul_cpu_kernel.cc44 auto values_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, VALUES); in InitKernel() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc498 ShapeVector values_shape = {Shape::SHP_ANY}; in InferImplCTCGreedyDecoder() local
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py6166 def infer_shape(self, sequence_shape, values_shape): argument