| /dports/misc/mxnet/incubator-mxnet-1.9.0/plugin/caffe/ |
| H A D | caffe_loss.cc | 55 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
59 aux_shape.resize(this->ListAuxiliaryStates().size()); in CreateOperatorEx()
61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | caffe_op.cc | 55 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
59 aux_shape.resize(this->ListAuxiliaryStates().size()); in CreateOperatorEx()
61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/plugin/caffe/ |
| H A D | caffe_loss.cc | 55 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
59 aux_shape.resize(this->ListAuxiliaryStates().size()); in CreateOperatorEx()
61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | caffe_op.cc | 55 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
59 aux_shape.resize(this->ListAuxiliaryStates().size()); in CreateOperatorEx()
61 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
| H A D | make_loss.cc | 39 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | convolution_v1.cc | 51 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
54 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | batch_norm_v1.cc | 39 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | pooling_v1.cc | 56 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
59 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
| H A D | make_loss.cc | 39 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | convolution_v1.cc | 51 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
54 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | batch_norm_v1.cc | 39 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
42 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | pooling_v1.cc | 56 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
59 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/include/mxnet/ |
| H A D | ndarray.h | 241 inline const mxnet::TShape& aux_shape(size_t index) const { in aux_shape() function
293 auto shape = aux_shape(i); in aux_data()
342 CHECK_EQ(aux_shape(rowsparse::kIdx)[0], storage_shape()[0]) in storage_initialized()
345 return aux_shape(rowsparse::kIdx).Size() != 0; in storage_initialized()
347 CHECK_EQ(aux_shape(csr::kIdx)[0], storage_shape()[0]) in storage_initialized()
349 << " vs. aux shape " << aux_shape(csr::kIdx); in storage_initialized()
350 return aux_shape(csr::kIdx).Size() != 0; in storage_initialized()
698 ptr_->CheckAndAllocAuxData(i, aux_shape); in CheckAndAllocAuxData()
1033 auto aux_shape = aux_shapes[rowsparse::kIdx]; in CheckAndAlloc() local
1034 CheckAndAllocAuxData(rowsparse::kIdx, aux_shape); in CheckAndAlloc()
[all …]
|
| H A D | operator.h | 202 mxnet::ShapeVector *aux_shape) const = 0;
260 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
264 aux_shape.resize(this->ListAuxiliaryStates().size()); in CreateOperatorEx()
266 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/include/mxnet/ |
| H A D | ndarray.h | 241 inline const mxnet::TShape& aux_shape(size_t index) const { in aux_shape() function
293 auto shape = aux_shape(i); in aux_data()
342 CHECK_EQ(aux_shape(rowsparse::kIdx)[0], storage_shape()[0]) in storage_initialized()
345 return aux_shape(rowsparse::kIdx).Size() != 0; in storage_initialized()
347 CHECK_EQ(aux_shape(csr::kIdx)[0], storage_shape()[0]) in storage_initialized()
349 << " vs. aux shape " << aux_shape(csr::kIdx); in storage_initialized()
350 return aux_shape(csr::kIdx).Size() != 0; in storage_initialized()
698 ptr_->CheckAndAllocAuxData(i, aux_shape); in CheckAndAllocAuxData()
1033 auto aux_shape = aux_shapes[rowsparse::kIdx]; in CheckAndAlloc() local
1034 CheckAndAllocAuxData(rowsparse::kIdx, aux_shape); in CheckAndAlloc()
[all …]
|
| H A D | operator.h | 202 mxnet::ShapeVector *aux_shape) const = 0;
260 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
264 aux_shape.resize(this->ListAuxiliaryStates().size()); in CreateOperatorEx()
266 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | deformable_convolution.cc | 33 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
36 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | modulated_deformable_convolution.cc | 31 std::vector<TShape> out_shape, aux_shape; in CreateOperatorEx() local
34 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | sync_batch_norm.cc | 38 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
41 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | count_sketch.cc | 36 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
39 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | deformable_convolution.cc | 33 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
36 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | modulated_deformable_convolution.cc | 31 std::vector<TShape> out_shape, aux_shape; in CreateOperatorEx() local
34 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| H A D | sync_batch_norm.cc | 38 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
41 CHECK(InferShape(in_shape, &out_shape, &aux_shape)); in CreateOperatorEx()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | elemwise_scatter_op.h | 147 DCHECK_LE(outputs[0].aux_shape(rowsparse::kIdx).Size(), in ScatterWrap()
148 inputs[0].aux_shape(rowsparse::kIdx).Size()); in ScatterWrap()
159 DCHECK_LE(outputs[0].aux_shape(rowsparse::kIdx).Size(), in ScatterWrap()
160 inputs[0].aux_shape(rowsparse::kIdx).Size()); in ScatterWrap()
189 CHECK_EQ(inputs[0].aux_shape(rowsparse::kIdx).Size(), in ComputeEx_()
190 outputs[0].aux_shape(rowsparse::kIdx).Size()); in ComputeEx_()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | elemwise_scatter_op.h | 147 DCHECK_LE(outputs[0].aux_shape(rowsparse::kIdx).Size(), in ScatterWrap()
148 inputs[0].aux_shape(rowsparse::kIdx).Size()); in ScatterWrap()
159 DCHECK_LE(outputs[0].aux_shape(rowsparse::kIdx).Size(), in ScatterWrap()
160 inputs[0].aux_shape(rowsparse::kIdx).Size()); in ScatterWrap()
189 CHECK_EQ(inputs[0].aux_shape(rowsparse::kIdx).Size(), in ComputeEx_()
190 outputs[0].aux_shape(rowsparse::kIdx).Size()); in ComputeEx_()
|