Home
last modified time | relevance | path

Searched refs:block_size_ (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetodepth_op.cc34 OP_REQUIRES_OK(ctx, ctx->GetAttr("block_size", &block_size_)); in SpaceToDepthOp()
36 ctx, block_size_ > 1, in SpaceToDepthOp()
37 errors::InvalidArgument("Block size should be > 1: ", block_size_)); in SpaceToDepthOp()
81 OP_REQUIRES(ctx, input_shape[1 + i] % block_size_ == 0, in Compile()
84 " is not divisible by block_size=", block_size_)); in Compile()
85 block_elems *= block_size_; in Compile()
90 reshaped_shape.push_back(input_shape[1 + i] / block_size_); in Compile()
91 reshaped_shape.push_back(block_size_); in Compile()
106 output_shape.push_back(input_shape[1 + i] / block_size_); in Compile()
113 OP_REQUIRES(ctx, input_shape[2 + i] % block_size_ == 0, in Compile()
[all …]
Ddepthtospace_op.cc34 OP_REQUIRES_OK(ctx, ctx->GetAttr("block_size", &block_size_)); in DepthToSpaceOp()
36 ctx, block_size_ > 1, in DepthToSpaceOp()
37 errors::InvalidArgument("Block size should be > 1: ", block_size_)); in DepthToSpaceOp()
85 reshaped_shape.push_back(block_size_); in Compile()
86 block_elems *= block_size_; in Compile()
99 output_shape.push_back(input_shape[1 + i] * block_size_); in Compile()
107 reshaped_shape.push_back(block_size_); in Compile()
108 block_elems *= block_size_; in Compile()
125 output_shape.push_back(input_shape[2 + i] * block_size_); in Compile()
140 input_shape[feature_dim] % (block_size_ * block_size_) == 0, in Compile()
[all …]
Dspacetobatch_op.cc171 OP_REQUIRES_OK(ctx, ctx->GetAttr("block_size", &block_size_)); in SpaceToBatchOp()
173 ctx, block_size_ > 1, in SpaceToBatchOp()
174 errors::InvalidArgument("Block size should be > 1: ", block_size_)); in SpaceToBatchOp()
182 {block_size_, block_size_}, paddings); in Compile()
186 int block_size_; member in tensorflow::__anona108e8f00111::SpaceToBatchOp
Dbatchtospace_op.cc169 OP_REQUIRES_OK(ctx, ctx->GetAttr("block_size", &block_size_)); in BatchToSpaceOp()
171 ctx, block_size_ > 1, in BatchToSpaceOp()
172 errors::InvalidArgument("Block size should be > 1: ", block_size_)); in BatchToSpaceOp()
180 {block_size_, block_size_}, crops); in Compile()
184 int block_size_; member in tensorflow::__anon5c1a54300111::BatchToSpaceOp
/external/tensorflow/tensorflow/core/kernels/
Dspacetodepth_op.cc67 OP_REQUIRES_OK(context, context->GetAttr("block_size", &block_size_)); in SpaceToDepthOp()
68 OP_REQUIRES(context, block_size_ > 1, in SpaceToDepthOp()
70 block_size_)); in SpaceToDepthOp()
109 (width % block_size_) == 0 && (height % block_size_) == 0, in Compute()
112 " should be divisible by block_size: ", block_size_)); in Compute()
116 const int output_depth = input_depth * block_size_ * block_size_; in Compute()
117 const int output_width = width / block_size_; in Compute()
118 const int output_height = height / block_size_; in Compute()
136 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_, in Compute()
142 block_size_, outputs_tensor->tensor<RT, 4>()); in Compute()
[all …]
Ddepthtospace_op.cc52 OP_REQUIRES_OK(context, context->GetAttr("block_size", &block_size_)); in DepthToSpaceOp()
53 OP_REQUIRES(context, block_size_ > 1, in DepthToSpaceOp()
55 block_size_)); in DepthToSpaceOp()
92 const int block_size_sq = block_size_ * block_size_; in Compute()
101 const int output_width = input_width * block_size_; in Compute()
102 const int output_height = input_height * block_size_; in Compute()
121 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_, in Compute()
126 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_, in Compute()
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
142 int block_size_; member in tensorflow::DepthToSpaceOp
Dspacetobatch_op.cc227 OP_REQUIRES_OK(context, context->GetAttr("block_size", &block_size_)); in SpaceToBatchOp()
229 context, block_size_ > 1, in SpaceToBatchOp()
230 errors::InvalidArgument("Block size should be > 1: ", block_size_)); in SpaceToBatchOp()
235 block_shape_vec(0) = block_size_; in SpaceToBatchOp()
236 block_shape_vec(1) = block_size_; in SpaceToBatchOp()
253 int block_size_; member in tensorflow::SpaceToBatchOp
/external/webrtc/webrtc/common_audio/
Dblocker.cc109 block_size_(block_size), in Blocker()
112 initial_delay_(block_size_ - gcd(chunk_size, shift_amount)), in Blocker()
116 input_block_(block_size_, num_input_channels_), in Blocker()
117 output_block_(block_size_, num_output_channels_), in Blocker()
118 window_(new float[block_size_]), in Blocker()
122 RTC_CHECK_LE(shift_amount_, block_size_); in Blocker()
124 memcpy(window_.get(), window, block_size_ * sizeof(*window_.get())); in Blocker()
182 block_size_); in ProcessChunk()
183 input_buffer_.MoveReadPositionBackward(block_size_ - shift_amount_); in ProcessChunk()
186 block_size_, in ProcessChunk()
[all …]
/external/brotli/c/enc/
Dmetablock_inc.h34 size_t block_size_;
61 self->block_size_ = 0; in FN()
89 self->block_size_ = in FN()
90 BROTLI_MAX(size_t, self->block_size_, self->min_block_size_); in FN()
93 split->lengths[0] = (uint32_t)self->block_size_; in FN()
103 self->block_size_ = 0; in FN()
104 } else if (self->block_size_ > 0) { in FN()
125 split->lengths[self->num_blocks_] = (uint32_t)self->block_size_; in FN()
136 self->block_size_ = 0; in FN()
141 split->lengths[self->num_blocks_] = (uint32_t)self->block_size_; in FN()
[all …]
Dmetablock.c328 size_t block_size_; member
357 self->block_size_ = 0; in InitContextBlockSplitter()
393 if (self->block_size_ < self->min_block_size_) { in ContextBlockSplitterFinishBlock()
394 self->block_size_ = self->min_block_size_; in ContextBlockSplitterFinishBlock()
399 split->lengths[0] = (uint32_t)self->block_size_; in ContextBlockSplitterFinishBlock()
414 self->block_size_ = 0; in ContextBlockSplitterFinishBlock()
415 } else if (self->block_size_ > 0) { in ContextBlockSplitterFinishBlock()
448 split->lengths[self->num_blocks_] = (uint32_t)self->block_size_; in ContextBlockSplitterFinishBlock()
463 self->block_size_ = 0; in ContextBlockSplitterFinishBlock()
468 split->lengths[self->num_blocks_] = (uint32_t)self->block_size_; in ContextBlockSplitterFinishBlock()
[all …]
/external/libtextclassifier/native/utils/base/
Darena.cc119 block_size_(orig_block_size), in BaseArena()
127 TC3_CHECK_GT(block_size_, kDefaultAlignment) in BaseArena()
145 TC3_CHECK_EQ(block_size_ & (kPageSize - 1), 0) << "block_size is not a" in BaseArena()
147 first_blocks_[0].mem = AllocateAlignedBytes(block_size_, kPageSize); in BaseArena()
151 first_blocks_[0].mem = AllocateBytes(block_size_); in BaseArena()
154 first_blocks_[0].size = block_size_; in BaseArena()
217 TC3_ARENASET(status_.bytes_allocated_ = block_size_); in Reset()
234 AllocatedBlock *block = AllocNewBlock(block_size_, alignment); in MakeNewBlock()
379 if (block_size_ == 0 || size > block_size_/4) { in AllocNewBlock()
452 assert(freestart_ >= last_alloc_ && freestart_ <= last_alloc_ + block_size_); in AllocNewBlock()
/external/tensorflow/tensorflow/core/platform/cloud/
Dram_file_block_cache.cc86 if (block->data.size() < block_size_) { in UpdateLRU()
133 block->data.resize(block_size_, 0); in MaybeFetch()
135 status.Update(block_fetcher_(key.first, key.second, block_size_, in MaybeFetch()
180 size_t start = block_size_ * (offset / block_size_); in Read()
181 size_t finish = block_size_ * ((offset + n) / block_size_); in Read()
183 finish += block_size_; in Read()
187 for (size_t pos = start; pos < finish; pos += block_size_) { in Read()
221 if (data.size() < block_size_) { in Read()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_stream.h77 std::tie(max_levels_, block_size_) = GetQuantileSpecs(eps, max_elements); in WeightedQuantilesStream()
78 buffer_ = Buffer(block_size_, max_elements); in WeightedQuantilesStream()
110 local_summary_.Compress(block_size_, eps_); in PushBuffer()
121 local_summary_.Compress(block_size_, eps_); in PushSummary()
264 local_summary_.Size() <= block_size_ + 1) { in PropagateLocalSummary()
269 local_summary_.Compress(block_size_, eps_); in PropagateLocalSummary()
280 int64 block_size_; variable
/external/tensorflow/tensorflow/core/lib/core/
Darena.cc47 block_size_(block_size), in Arena()
54 reinterpret_cast<char*>(port::AlignedMalloc(block_size_, sizeof(void*))); in Arena()
56 first_blocks_[0].size = block_size_; in Arena()
111 AllocatedBlock* block = AllocNewBlock(block_size_, alignment); in MakeNewBlock()
204 if (block_size_ == 0 || size > block_size_ / 4) { in GetMemoryFallback()
/external/webrtc/webrtc/modules/audio_coding/neteq/
Dtime_stretch_unittest.cc65 block_size_(30 * sample_rate_hz_ / 1000), // 30 ms in TimeStretchTest()
66 audio_(new int16_t[block_size_]), in TimeStretchTest()
72 RTC_CHECK(input_file_->Read(block_size_, audio_.get())); in Next30Ms()
84 UpdateReturnStats(accelerate.Process(Next30Ms(), block_size_, fast_mode, in TestAccelerate()
105 const size_t block_size_; member in webrtc::TimeStretchTest
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconvolution_transposed_3d.cc359 block_size_(2, 2, 1, 2) {} in ConvolutionTransposed3D()
374 block_size_(operation.block_size_), in ConvolutionTransposed3D()
391 std::swap(block_size_, operation.block_size_); in operator =()
403 block_size_, linked_operations_); in Compile()
406 if (creation_context.device->IsPowerVR() && block_size_.y != 1) { in Compile()
444 IntegralDivideRoundUp(dst_[0]->Slices(), block_size_.w))); in BindArguments()
451 const int aligned_w = AlignByN(dst_[0]->Width(), stride_.x * block_size_.x); in GetGridSize()
452 const int aligned_h = AlignByN(dst_[0]->Height(), stride_.y * block_size_.y); in GetGridSize()
453 const int aligned_d = AlignByN(dst_[0]->Depth(), stride_.z * block_size_.z); in GetGridSize()
455 IntegralDivideRoundUp(aligned_w, block_size_.x) * dst_[0]->Batch(); in GetGridSize()
[all …]
Dconvolution_transposed.h84 int3 block_size_ = int3(1, 1, 1); variable
94 AlignByN(IntegralDivideRoundUp(weights.shape.o, 4), block_size_.z); in UploadWeights()
163 AlignByN(IntegralDivideRoundUp(weights.shape.o, 4), block_size_.z); in RearrangeWeightsData()
171 for (int d = 0; d < dst_depth / block_size_.z; ++d) { in RearrangeWeightsData()
175 for (int sub_d = 0; sub_d < block_size_.z; ++sub_d) { in RearrangeWeightsData()
180 const int d_ch = (d * block_size_.z + sub_d) * 4 + i; in RearrangeWeightsData()
196 int x_coord = d * block_size_.z + sub_d; in RearrangeWeightsData()
Dconvolution_transposed_3d.h84 int4 block_size_ = int4(1, 1, 1, 1); // WHDS variable
94 AlignByN(IntegralDivideRoundUp(weights.shape.o, 4), block_size_.z); in UploadWeights()
165 AlignByN(IntegralDivideRoundUp(weights.shape.o, 4), block_size_.w); in RearrangeWeightsData()
174 for (int d = 0; d < dst_depth / block_size_.w; ++d) { in RearrangeWeightsData()
179 for (int sub_d = 0; sub_d < block_size_.w; ++sub_d) { in RearrangeWeightsData()
184 const int d_ch = (d * block_size_.w + sub_d) * 4 + i; in RearrangeWeightsData()
200 int x_coord = d * block_size_.w + sub_d; in RearrangeWeightsData()
Dconvolution_transposed.cc307 block_size_(2, 2, 2) {} in ConvolutionTransposed()
321 block_size_(operation.block_size_), in ConvolutionTransposed()
338 std::swap(block_size_, operation.block_size_); in operator =()
349 block_size_, linked_operations_); in Compile()
381 const int aligned_w = AlignByN(dst_[0]->Width(), stride_.x * block_size_.x); in GetGridSize()
382 const int aligned_h = AlignByN(dst_[0]->Height(), stride_.y * block_size_.y); in GetGridSize()
384 IntegralDivideRoundUp(aligned_w, block_size_.x) * dst_[0]->Batch(); in GetGridSize()
385 const int grid_y = IntegralDivideRoundUp(aligned_h, block_size_.y); in GetGridSize()
386 const int grid_z = IntegralDivideRoundUp(dst_[0]->Slices(), block_size_.z); in GetGridSize()
Dconv_texture.h95 int3 block_size_ = int3(2, 2, 2); variable
118 AlignByN(IntegralDivideRoundUp(weights.shape.o, 4), block_size_.z); in UploadWeights()
174 AlignByN(IntegralDivideRoundUp(weights.shape.o, 4), block_size_.z); in RearrangeWeightsData()
179 for (int d = 0; d < dst_depth / block_size_.z; ++d) { in RearrangeWeightsData()
183 for (int sub_d = 0; sub_d < block_size_.z; ++sub_d) { in RearrangeWeightsData()
188 const int d_ch = (d * block_size_.z + sub_d) * 4 + i; in RearrangeWeightsData()
198 int x_coord = d * block_size_.z + sub_d; in RearrangeWeightsData()
/external/libaom/libaom/test/
Dhadamard_test.cc138 block_size_ = bwh_ * bwh_; in SetUp()
164 for (int i = 0; i < block_size_; ++i) a[i] = Rand(); in CompareReferenceRandom()
170 std::sort(b, b + block_size_); in CompareReferenceRandom()
171 std::sort(b_ref, b_ref + block_size_); in CompareReferenceRandom()
180 for (int i = 0; i < block_size_ * 8; ++i) a[i] = Rand(); in VaryStride()
191 std::sort(b, b + block_size_); in VaryStride()
192 std::sort(b_ref, b_ref + block_size_); in VaryStride()
220 int block_size_; member in __anon54c7ca360111::HadamardTestBase
/external/grpc-grpc/include/grpcpp/impl/codegen/
Dproto_buffer_writer.h61 : block_size_(block_size), in ProtoBufferWriter()
101 remain > static_cast<size_t>(block_size_) ? block_size_ : remain; in Next()
154 const int block_size_; ///< size to alloc for each new \a grpc_slice needed variable
/external/webrtc/webrtc/common_audio/resampler/
Dsinc_resampler.cc178 assert(block_size_ > kKernelSize); in SincResampler()
198 block_size_ = r4_ - r2_; in UpdateRegions()
290 ceil((block_size_ - virtual_source_idx_) / current_io_ratio)); in Resample()
292 assert(virtual_source_idx_ < block_size_); in Resample()
330 virtual_source_idx_ -= block_size_; in Resample()
348 return static_cast<size_t>(block_size_ / io_sample_rate_ratio_); in ChunkSize()
/external/libvpx/libvpx/test/
Dhadamard_test.cc136 block_size_ = bwh_ * bwh_; in SetUp()
162 for (int i = 0; i < block_size_; ++i) a[i] = Rand(); in CompareReferenceRandom()
168 std::sort(b, b + block_size_); in CompareReferenceRandom()
169 std::sort(b_ref, b_ref + block_size_); in CompareReferenceRandom()
178 for (int i = 0; i < block_size_ * 8; ++i) a[i] = Rand(); in VaryStride()
189 std::sort(b, b + block_size_); in VaryStride()
190 std::sort(b_ref, b_ref + block_size_); in VaryStride()
216 int block_size_; member in __anon932007980111::HadamardTestBase
/external/libgav1/libgav1/src/utils/
Dparameter_tree.h67 BlockSize block_size() const { return block_size_; } in block_size()
80 : row4x4_(row4x4), column4x4_(column4x4), block_size_(block_size) {} in ParameterTree()
86 BlockSize block_size_ = kBlockInvalid; variable

12