Home
last modified time | relevance | path

Searched refs:Slice (Results 1 – 25 of 347) sorted by relevance

12345678910>>...14

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.h49 CudnnBatchNormForwardInferenceThunk(const BufferAllocation::Slice& operand,
50 const BufferAllocation::Slice& scale,
51 const BufferAllocation::Slice& offset,
52 const BufferAllocation::Slice& mean,
53 const BufferAllocation::Slice& variance,
55 const BufferAllocation::Slice& output,
68 BufferAllocation::Slice operand_;
69 BufferAllocation::Slice scale_;
70 BufferAllocation::Slice offset_;
71 BufferAllocation::Slice mean_;
[all …]
Dcudnn_batchnorm_thunk.cc79 const BufferAllocation::Slice& operand, in CudnnBatchNormForwardInferenceThunk()
80 const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset, in CudnnBatchNormForwardInferenceThunk()
81 const BufferAllocation::Slice& mean, in CudnnBatchNormForwardInferenceThunk()
82 const BufferAllocation::Slice& variance, float epsilon, int64 feature_index, in CudnnBatchNormForwardInferenceThunk()
83 const BufferAllocation::Slice& output, const HloInstruction* hlo) in CudnnBatchNormForwardInferenceThunk()
136 const BufferAllocation::Slice& operand, in CudnnBatchNormForwardTrainingThunk()
137 const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset, in CudnnBatchNormForwardTrainingThunk()
139 const BufferAllocation::Slice& output_data, in CudnnBatchNormForwardTrainingThunk()
140 const BufferAllocation::Slice& output_mean, in CudnnBatchNormForwardTrainingThunk()
141 const BufferAllocation::Slice& output_inv_stddev, in CudnnBatchNormForwardTrainingThunk()
[all …]
Dconvolution_thunk.h47 std::vector<BufferAllocation::Slice> operand_slices,
48 BufferAllocation::Slice result_slice,
49 BufferAllocation::Slice scratch_slice,
50 BufferAllocation::Slice tuple_result_slice);
62 std::vector<BufferAllocation::Slice> operand_buffers_;
63 BufferAllocation::Slice result_buffer_;
64 BufferAllocation::Slice scratch_buffer_;
65 BufferAllocation::Slice tuple_result_buffer_;
Dcopy_thunk.h37 const BufferAllocation::Slice& destination_buffer,
49 const BufferAllocation::Slice destination_buffer_;
59 DeviceToDeviceCopyThunk(const BufferAllocation::Slice& source_buffer,
60 const BufferAllocation::Slice& destination_buffer,
72 const BufferAllocation::Slice source_buffer_;
73 const BufferAllocation::Slice destination_buffer_;
Dcholesky_thunk.h45 BufferAllocation::Slice a_buffer,
46 BufferAllocation::Slice workspace_buffer,
47 BufferAllocation::Slice info_buffer,
61 const BufferAllocation::Slice a_buffer_;
62 const BufferAllocation::Slice workspace_buffer_;
63 const BufferAllocation::Slice info_buffer_;
Dgemm_thunk.h40 GemmThunk(const BufferAllocation::Slice& lhs_buffer,
41 const BufferAllocation::Slice& rhs_buffer,
42 const BufferAllocation::Slice& output_buffer,
65 const BufferAllocation::Slice lhs_buffer_;
66 const BufferAllocation::Slice rhs_buffer_;
67 const BufferAllocation::Slice output_buffer_;
Dnccl_all_reduce_thunk.h44 const BufferAllocation::Slice& source_buffer,
45 const BufferAllocation::Slice& destination_buffer,
55 const BufferAllocation::Slice source_buffer_;
56 const BufferAllocation::Slice destination_buffer_;
Dtuple_thunk.h37 TupleThunk(absl::Span<const BufferAllocation::Slice> tuple_element_buffers, in TupleThunk()
38 const BufferAllocation::Slice& dest_buffer, in TupleThunk()
53 const std::vector<BufferAllocation::Slice> tuple_element_buffers_;
54 const BufferAllocation::Slice dest_buffer_;
Dconditional_thunk.h46 const BufferAllocation::Slice& branch_index_buffer_index,
47 absl::Span<const BufferAllocation::Slice> branch_operand_buffer_indexes,
62 BufferAllocation::Slice branch_index_buffer_index_;
63 std::vector<BufferAllocation::Slice> branch_operand_buffer_indexes_;
Dmemset_thunk.h35 explicit MemzeroThunk(const BufferAllocation::Slice& dest, in MemzeroThunk()
44 const BufferAllocation::Slice dest_;
52 const BufferAllocation::Slice& dest, in Memset32BitValueThunk()
62 const BufferAllocation::Slice dest_;
Dtriangular_solve_thunk.h42 const BufferAllocation::Slice& a_buffer,
43 const BufferAllocation::Slice& b_buffer,
61 const BufferAllocation::Slice a_buffer_;
62 const BufferAllocation::Slice b_buffer_;
/external/grpc-grpc/include/grpcpp/impl/codegen/
Dslice.h35 class Slice final {
38 Slice() : slice_(g_core_codegen_interface->grpc_empty_slice()) {} in Slice() function
40 ~Slice() { g_core_codegen_interface->grpc_slice_unref(slice_); } in ~Slice()
44 Slice(grpc_slice slice, AddRef) in Slice() function
49 Slice(grpc_slice slice, StealRef) : slice_(slice) {} in Slice() function
52 Slice(size_t len) in Slice() function
56 Slice(const void* buf, size_t len) in Slice() function
61 Slice(const grpc::string& str) in Slice() function
68 Slice(const void* buf, size_t len, StaticSlice) in Slice() function
73 Slice(const Slice& other) in Slice() function
[all …]
/external/grpc-grpc/test/cpp/util/
Dslice_test.cc36 void CheckSliceSize(const Slice& s, const grpc::string& content) { in CheckSliceSize()
39 void CheckSlice(const Slice& s, const grpc::string& content) { in CheckSlice()
47 Slice empty_slice; in TEST_F()
52 Slice sized_slice(strlen(kContent)); in TEST_F()
57 Slice spp(kContent); in TEST_F()
62 Slice spp(kContent, strlen(kContent)); in TEST_F()
67 Slice spp(kContent, strlen(kContent), Slice::STATIC_SLICE); in TEST_F()
74 Slice spp(x, strlen(x), [](void* p) { delete[] static_cast<char*>(p); }); in TEST_F()
79 Slice spp(const_cast<char*>(kContent), strlen(kContent), [](void* p) {}); in TEST_F()
91 Slice spp(t->x, strlen(t->x), in TEST_F()
[all …]
Dbyte_buffer_test.cc49 Slice s(kContent1); in TEST_F()
55 std::vector<Slice> slices; in TEST_F()
63 Slice s(kContent1); in TEST_F()
70 std::vector<Slice> slices; in TEST_F()
77 bool SliceEqual(const Slice& a, grpc_slice b) { in SliceEqual()
92 std::vector<Slice> slices; in TEST_F()
93 slices.push_back(Slice(hello, Slice::STEAL_REF)); in TEST_F()
94 slices.push_back(Slice(world, Slice::STEAL_REF)); in TEST_F()
105 std::vector<Slice> slices; in TEST_F()
106 slices.push_back(Slice(hello, Slice::STEAL_REF)); in TEST_F()
[all …]
/external/perfetto/include/perfetto/tracing/core/
Dslice.h33 struct Slice { struct
34 Slice() : start(nullptr), size(0) {} in Slice() function
35 Slice(const void* st, size_t sz) : start(st), size(sz) {} in Slice() argument
38 explicit Slice(std::unique_ptr<std::string> str) in Slice() argument
41 Slice(Slice&& other) noexcept = default;
44 static Slice Allocate(size_t size) { in Allocate() argument
45 Slice slice; in Allocate()
61 Slice(const Slice&) = delete; argument
62 void operator=(const Slice&) = delete; argument
71 using Slices = std::vector<Slice>; argument
/external/tensorflow/tensorflow/core/kernels/
Dslice_op_gpu.cu.cc31 template struct functor::Slice<GPUDevice, T, 1>; \
32 template struct functor::Slice<GPUDevice, T, 2>; \
33 template struct functor::Slice<GPUDevice, T, 3>; \
34 template struct functor::Slice<GPUDevice, T, 4>; \
35 template struct functor::Slice<GPUDevice, T, 5>; \
36 template struct functor::Slice<GPUDevice, T, 6>; \
37 template struct functor::Slice<GPUDevice, T, 7>;
/external/perfetto/src/protozero/
Dscattered_heap_buffer.cc23 ScatteredHeapBuffer::Slice::Slice(size_t size) in Slice() function in protozero::ScatteredHeapBuffer::Slice
33 ScatteredHeapBuffer::Slice::Slice(Slice&& slice) noexcept = default;
35 ScatteredHeapBuffer::Slice::~Slice() = default;
/external/perfetto/include/perfetto/protozero/
Dscattered_heap_buffer.h35 class PERFETTO_EXPORT Slice {
37 explicit Slice(size_t size);
38 Slice(Slice&& slice) noexcept;
39 ~Slice();
73 const std::vector<Slice>& slices() const { return slices_; } in slices()
89 std::vector<Slice> slices_;
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc36 BufferAllocation::Slice buffer_slice; in AddAliasingInformationToIrArray()
41 buffer_slice = BufferAllocation::Slice(kParameterAllocation, 0, 0); in AddAliasingInformationToIrArray()
43 const std::set<BufferAllocation::Slice> slices = in AddAliasingInformationToIrArray()
109 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain) { in GetAliasScopeMetadataForBuffer()
124 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain, in GetNoaliasMetadataForBuffer()
165 std::set<BufferAllocation::Slice> buffers; in GetNoaliasMetadataForBuffer()
172 const BufferAllocation::Slice noalias_slice = in GetNoaliasMetadataForBuffer()
198 for (const BufferAllocation::Slice noalias_slice : buffers) { in GetNoaliasMetadataForBuffer()
Dalias_analysis.h49 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain);
60 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain,
79 absl::flat_hash_map<BufferAllocation::Slice, llvm::MDNode*>
84 absl::flat_hash_map<BufferAllocation::Slice, llvm::MDNode*> noalias_metadata_;
/external/tensorflow/tensorflow/compiler/xla/tests/
Dslice_test.cc47 Slice(original, {0, 0, 0}, {3, 3, 1}, {1, 1, 1}); in TEST_F()
60 Slice(original, {0, 0, 0}, {3, 1, 3}, {1, 1, 1}); in TEST_F()
73 Slice(original, {0, 0, 0}, {1, 3, 3}, {1, 1, 1}); in TEST_F()
83 Slice(original, {0, 0}, {0, 0}, {1, 1}); in XLA_TEST_F()
91 Slice(original, {0, 15}, {0, 20}, {1, 1}); in XLA_TEST_F()
99 Slice(original, {1, 0}, {3, 0}, {1, 1}); in XLA_TEST_F()
114 Slice(original, {128, 128}, {256, 256}, {1, 1}); in XLA_TEST_F()
132 Slice(original, {0, 3072}, {1, 4096}, {1, 1}); in TEST_F()
153 Slice(original, {0, 0}, {16, 2}, {1, 1}); in TEST_F()
165 Slice(original, {1, 0, 8, 0}, {2, 2, 16, 128}, {1, 1, 1, 1}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h162 class Slice {
164 Slice() {} in Slice() function
165 Slice(const BufferAllocation* allocation, int64 offset, int64 size) in Slice() function
173 bool operator==(const Slice& other) const {
177 bool operator!=(const Slice& other) const { return !(*this == other); }
178 bool operator<(const Slice& other) const {
186 bool OverlapsWith(const Slice& other) const { in OverlapsWith()
194 friend H AbslHashValue(H h, const Slice& s) { in AbslHashValue()
209 Slice GetSlice(const LogicalBuffer& buffer) const;
347 std::ostream& operator<<(std::ostream& out, const BufferAllocation::Slice& s);
[all …]
Ddynamic_index_splitter_test.cc54 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
55 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
56 op::Reshape(op::Slice(op::Parameter(1))))); in TEST_F()
92 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
93 op::Reshape(op::Slice(op::Parameter(1))), in TEST_F()
94 op::Reshape(op::Slice(op::Parameter(1))))); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcross_op.cc62 auto u1 = xla::Slice(in0, starts, limits, strides); in Compile()
63 auto v1 = xla::Slice(in1, starts, limits, strides); in Compile()
66 auto u2 = xla::Slice(in0, starts, limits, strides); in Compile()
67 auto v2 = xla::Slice(in1, starts, limits, strides); in Compile()
70 auto u3 = xla::Slice(in0, starts, limits, strides); in Compile()
71 auto v3 = xla::Slice(in1, starts, limits, strides); in Compile()
/external/flatbuffers/net/FlatBuffers/
DByteBuffer.cs241 … return MemoryMarshal.Cast<byte, T>(_buffer.ReadOnlySpan.Slice(pos)).Slice(0, len).ToArray();
267 return _buffer.ReadOnlyMemory.Slice(pos, len); in ToReadOnlyMemory()
272 return _buffer.Memory.Slice(pos, len); in ToMemory()
277 return _buffer.Span.Slice(pos, len); in ToSpan()
392 Span<byte> span = _buffer.Span.Slice(offset, count); in PutByte()
455 Span<byte> span = _buffer.Span.Slice(offset); in PutUshort()
476 Span<byte> span = _buffer.Span.Slice(offset); in PutUint()
497 Span<byte> span = _buffer.Span.Slice(offset); in PutUlong()
633 fixed (byte* buffer = &MemoryMarshal.GetReference(_buffer.ReadOnlySpan.Slice(startPos))) in GetStringUTF8()
656 ReadOnlySpan<byte> span = _buffer.ReadOnlySpan.Slice(offset); in GetUshort()
[all …]

12345678910>>...14