Home
last modified time | relevance | path

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

123

/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 …]
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::__anon7c3f680c0111::BatchToSpaceOp
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::__anonc12dfccc0111::SpaceToBatchOp
/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<Device>(), 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<Device>(), Tinput_v, block_size_, in Compute()
126 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
136 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
141 int block_size_; member in tensorflow::DepthToSpaceOp
/external/webrtc/modules/audio_coding/codecs/opus/test/
Dblocker.cc107 block_size_(block_size), in Blocker()
110 initial_delay_(block_size_ - gcd(chunk_size, shift_amount)), in Blocker()
114 input_block_(block_size_, num_input_channels_), in Blocker()
115 output_block_(block_size_, num_output_channels_), in Blocker()
116 window_(new float[block_size_]), in Blocker()
120 RTC_CHECK_LE(shift_amount_, block_size_); in Blocker()
122 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()
185 ApplyWindow(window_.get(), block_size_, num_input_channels_, 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 …]
/external/pigweed/pw_trace_tokenized/
Dtrace_buffer.cc38 buffer->block_size_ = 0; // Skip this block in TraceSinkStartBlock()
41 buffer->block_size_ = static_cast<uint16_t>(size); in TraceSinkStartBlock()
49 if (buffer->block_size_ == 0 || in TraceSinkAddBytes()
50 buffer->block_idx_ + size > buffer->block_size_) { in TraceSinkAddBytes()
59 if (buffer->block_idx_ != buffer->block_size_) { in TraceSinkEndBlock()
63 &buffer->current_block_[0], buffer->block_size_)); in TraceSinkEndBlock()
71 uint16_t block_size_ = 0; member in pw::trace::__anon76b02f780111::TraceBuffer
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed.cc37 block_size_(2, 2, 1, 2) { in ConvolutionTransposed()
54 block_size_ = is_f16 ? int4(2, 1, 1, 2) : int4(2, 1, 1, 1); in ConvolutionTransposed()
56 block_size_ = is_f16 ? int4(2, 2, 1, 2) : int4(2, 2, 1, 1); in ConvolutionTransposed()
62 block_size_.y *= block_size_.w; in ConvolutionTransposed()
64 block_size_.w = 1; in ConvolutionTransposed()
74 weights_are_buffer, block_size_); in ConvolutionTransposed()
83 block_size_(2, 2, 1, 2) { in ConvolutionTransposed()
100 block_size_ = is_f16 ? int4(2, 1, 1, 2) : int4(2, 1, 1, 1); in ConvolutionTransposed()
102 block_size_ = is_f16 ? int4(2, 2, 1, 2) : int4(2, 2, 1, 1); in ConvolutionTransposed()
108 block_size_.y *= block_size_.w; in ConvolutionTransposed()
[all …]
Dconvolution_transposed.h58 desc.output_group_size = block_size_.w; in GetWeightsDescription()
93 int4 block_size_ = int4(1, 1, 1, 1); // WHDS variable
120 AlignByN(DivideRoundUp(weights.shape.o, 4), block_size_.w); in UploadWeights()
144 AlignByN(DivideRoundUp(weights.shape.o, 4), block_size_.w); in UploadWeights()
160 RearrangeWeightsToODHWIOGroupI4O4(weights, block_size_.w, in UploadWeights()
163 RearrangeWeightsToI4DHWIOOGroupO4(weights, block_size_.w, in UploadWeights()
169 RearrangeWeightsToODHWIOGroupI4O4(weights, block_size_.w, in UploadWeights()
172 RearrangeWeightsToI4DHWIOOGroupO4(weights, block_size_.w, in UploadWeights()
/external/libtextclassifier/native/utils/base/
Darena.cc121 block_size_(orig_block_size), in BaseArena()
129 TC3_CHECK_GT(block_size_, kDefaultAlignment) in BaseArena()
147 TC3_CHECK_EQ(block_size_ & (kPageSize - 1), 0) << "block_size is not a" in BaseArena()
149 first_blocks_[0].mem = AllocateAlignedBytes(block_size_, kPageSize); in BaseArena()
153 first_blocks_[0].mem = AllocateBytes(block_size_); in BaseArena()
156 first_blocks_[0].size = block_size_; in BaseArena()
219 TC3_ARENASET(status_.bytes_allocated_ = block_size_); in Reset()
236 AllocatedBlock *block = AllocNewBlock(block_size_, alignment); in MakeNewBlock()
381 if (block_size_ == 0 || size > block_size_/4) { in AllocNewBlock()
454 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/c/experimental/filesystem/plugins/gcs/
Dram_file_block_cache.cc87 if (block->data.size() < block_size_) { in UpdateLRU()
135 block->data.resize(block_size_, 0); in MaybeFetch()
137 bytes_transferred = 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()
227 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/modules/audio_coding/neteq/
Dtime_stretch_unittest.cc64 block_size_(30 * sample_rate_hz_ / 1000), // 30 ms in TimeStretchTest()
65 audio_(new int16_t[block_size_]), in TimeStretchTest()
69 RTC_CHECK(input_file_->Read(block_size_, audio_.get())); in Next30Ms()
81 UpdateReturnStats(accelerate.Process(Next30Ms(), block_size_, fast_mode, in TestAccelerate()
102 const size_t block_size_; member in webrtc::TimeStretchTest
/external/icing/icing/index/main/
Dflash-index-storage-header.h86 if (size() + sizeof(Header::IndexBlockInfo) > block_size_) { in AddIndexBlockInfo()
101 return filesystem_->PWrite(fd, 0, header_buffer_.get(), block_size_); in Write()
109 block_size_(block_size) {} in HeaderBlock()
113 int block_size_; variable
/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/modules/audio_processing/
Drms_level.cc57 block_size_ = absl::nullopt; in Reset()
117 ComputeRms(max_sum_square_ / *block_size_)}; in AverageAndPeak()
123 if (block_size_ != block_size) { in CheckBlockSize()
125 block_size_ = block_size; in CheckBlockSize()
/external/webrtc/common_audio/resampler/
Dsinc_resampler.cc175 RTC_DCHECK_GT(block_size_, kKernelSize); in SincResampler()
195 block_size_ = r4_ - r2_; in UpdateRegions()
287 ceil((block_size_ - virtual_source_idx_) / current_io_ratio)); in Resample()
289 RTC_DCHECK_LT(virtual_source_idx_, block_size_); in Resample()
327 virtual_source_idx_ -= block_size_; in Resample()
345 return static_cast<size_t>(block_size_ / io_sample_rate_ratio_); in ChunkSize()
/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 __anonc98e30920111::HadamardTestBase
/external/rust/crates/grpcio-sys/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/rust/crates/grpcio-sys/grpc/spm-cpp-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/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 __anon07e66df40111::HadamardTestBase

123