Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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
H A 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
H A 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
H A Ddequantize_op.cc102 int num_slices = 1; in Compute() local
H A Dquantize_op.cc114 int num_slices = 1; in Compute() local
/aosp_15_r20/external/grpc-grpc/test/core/http/
H A Dparser_test.cc44 size_t num_slices; in test_request_succeeds() local
102 size_t num_slices; in test_succeeds() local
156 size_t num_slices; in test_fails() local
188 size_t num_slices; in test_request_fails() local
/aosp_15_r20/external/grpc-grpc/test/cpp/microbenchmarks/
H A Dbm_byte_buffer.cc39 int num_slices = state.range(0); in BM_ByteBuffer_Copy() local
56 const int num_slices = state.range(0); in BM_ByteBufferReader_Next() local
84 const int num_slices = state.range(0); in BM_ByteBufferReader_Peek() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dgroup_assignment.h77 int num_slices() { return num_slices_; } in num_slices() function
113 int num_slices() const { return slices_.size(); } in num_slices() function
H A Dgroup_assignment.cc41 GroupAssignment::ReplicaToDeviceMap::DefaultReplicaToDeviceMap(int num_slices, in DefaultReplicaToDeviceMap()
205 int num_slices = replica_to_device_map_.num_slices(); in GlobalToSlices() local
H A Dgroup_assignment_test.cc56 const std::vector<std::vector<int>>& replica_ids, int num_slices, in CreateGroupAssignment()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_util_test.cc39 grpc::ByteBuffer MakeBuffer(const string& str, int num_slices) { in MakeBuffer()
184 const int num_slices = state.range(1); in BM_ParseGrpc() local
H A Dgrpc_tensor_coding.cc216 int num_slices = 0; in EncodeTensorToByteBuffer() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorTopK.cpp43 int64_t num_slices = self.numel() / slice_size; in should_use_sort() local
H A DTensorTopK.cu609 int get_items_per_thread(uint64_t num_slices, uint64_t slice_size) { in get_items_per_thread()
773 bool should_use_multiblock(int64_t num_slices, int64_t slice_size) { in should_use_multiblock()
H A DSortingCommon.cuh135 int64_t num_slices = 1; in run_launcher() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DUnstackFixture.h69 const unsigned int num_slices = std::min(num, axis_size); in compute_target() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEUnstack.cpp99 …const unsigned int num_slices = std::min(output_vector.size(), input->dimension(wrap_axis(axis, in… in validate() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLUnstack.cpp104 …const unsigned int num_slices = std::min(output_vector.size(), input->dimension(wrap_axis(axis, in… in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dsharding_util_ops.cc51 int& num_slices, in GetAndValidateAttributesHelper()
108 int& num_slices, std::vector<int32>& paddings, in GetAndValidateAttributes()
/aosp_15_r20/external/perfetto/src/protozero/filtering/
H A Dmessage_filter.cc101 size_t num_slices) { in FilterMessageFragments()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dtensor_slice_dataset_op_test.cc279 int64_t num_slices = params.num_slices(); in TEST_P() local
H A Dtensor_slice_dataset_op.cc234 const int64_t num_slices = inputs[0].dim_size(0); in MakeDataset() local
H A Dsparse_tensor_slice_dataset_op_test.cc380 int64_t num_slices = dataset_->Cardinality(); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_slice_reader.cc319 const int num_slices = e.second->Slices().size(); in DebugString() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DConv.cpp307 int8_t num_slices = 4 - num_input_channels % 4; // number of slices we need to pad in apply_impl() local

12