Searched defs:array_input (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | kldiv.cc | 72 Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1> > array_input(input + begin, length, 1); in ComputeKLDivKernel() local
84 Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1> > array_input(input, total, 1); in ComputeKLDivKernel() local
|
| D | kldivlossgrad.cc | 199 Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1> > array_input(input + begin, length, 1); in KlDivLossGradCompute() local
216 Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1> > array_input(input, input_total, 1); in KlDivLossGradCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/ |
| D | uniform_candidate_sampler_gpu_kernel.cc | 75 auto array_input = std::vector<T>(input_size_, kIndex0); in LaunchKernel() local
|
| /third_party/protobuf/src/google/protobuf/io/ |
| D | zero_copy_stream_unittest.cc | 973 ArrayInputStream array_input(buffer, kBufferSize); in TEST_F() local
987 ArrayInputStream array_input(buffer, kBufferSize, kHalfBufferSize); in TEST_F() local
|