Home
last modified time | relevance | path

Searched defs:num_slices (Results 1 – 25 of 28) sorted by relevance

12

/external/grpc-grpc/test/core/http/
Dparser_test.cc41 size_t num_slices; in test_request_succeeds() local
99 size_t num_slices; in test_succeeds() local
153 size_t num_slices; in test_fails() local
186 size_t num_slices; in test_request_fails() local
/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op_test.cc112 int num_slices = (axis == -1) ? 1 : dims[axis]; in ScalePerSliceAlongAxis() local
142 const int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
193 const int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
247 const int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
330 const int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
385 const int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
Dquantize_op_test.cc76 int num_slices = (axis == -1) ? 1 : dims[axis]; in ScalePerSliceAlongAxis() local
97 int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
178 int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
228 int num_slices = (axis == -1) ? 1 : dims[axis]; in TEST_P() local
Ddequantize_op_test.cc142 int num_slices = (axis == -1) ? 1 : dims[axis]; in ScalePerSliceAlongAxis() local
156 int num_slices = (axis == -1) ? 1 : dims[axis]; in RunDequantizeScaledTest() local
Ddequantize_op.cc97 int num_slices = 1; in Compute() local
Dquantize_op.cc114 int num_slices = 1; in Compute() local
Dsparse_matmul_op.cc1297 const MatrixR& mat, const int num_rows, const int num_slices, in SliceMatrix()
1315 const int num_slices = (num_cols + N - 1) / N; in CreateDenseSlices() local
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_util_test.cc36 grpc::ByteBuffer MakeBuffer(const string& str, int num_slices) { in MakeBuffer()
163 const int num_slices = state.range(1); in BM_ParseGrpc() local
Dgrpc_tensor_coding.cc216 int num_slices = 0; in EncodeTensorToByteBuffer() local
/external/ComputeLibrary/tests/validation/fixtures/
DUnstackFixture.h69 const unsigned int num_slices = std::min(num, axis_size); in compute_target() local
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEUnstack.cpp97 …const unsigned int num_slices = std::min(output_vector.size(), input->dimension(wrap_axis(axis, in… in validate() local
/external/ComputeLibrary/src/runtime/CL/functions/
DCLUnstack.cpp101 …const unsigned int num_slices = std::min(output_vector.size(), input->dimension(wrap_axis(axis, in… in validate() local
/external/perfetto/src/protozero/filtering/
Dmessage_filter.cc98 size_t num_slices) { in FilterMessageFragments()
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op.cc194 const int64_t num_slices = inputs[0].dim_size(0); in MakeDataset() local
Dtensor_slice_dataset_op_test.cc279 int64_t num_slices = params.num_slices(); in TEST_P() local
Dsparse_tensor_slice_dataset_op_test.cc380 int64_t num_slices = dataset_->Cardinality(); in TEST_P() local
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.cc312 const int num_slices = e.second->Slices().size(); in DebugString() local
Dbatch_util.cc218 int64_t dst_offset, int64_t num_slices, in CopyContiguousSlices()
/external/mesa3d/src/intel/dev/
Dgen_device_info.h125 unsigned num_slices; member
/external/tensorflow/tensorflow/lite/kernels/
Drfft2d.cc371 int num_slices = 1; in Rfft2dHelper() local
Dirfft2d.cc319 int num_slices = 1; in Irfft2dHelper() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc522 auto num_slices = slices_to_concatenate[j][i].shape().dim_size(0); in GetNextInternal() local
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_texture.c95 unsigned align_x, align_y, nblocksx, nblocksy, block_size, num_slices; in llvmpipe_texture_layout() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc51 int& num_slices, std::vector<int32>& paddings, in GetAndValidateAttributes()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.h307 int64 num_slices() const { return components_[0].dim_size(0); } in num_slices() function

12