Home
last modified time | relevance | path

Searched refs:shape_ptr (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.cc156 const int64* const shape_ptr = shape_.data(); in IndicesValidMatrix32BitFastPath() local
161 DCHECK_LE(shape_ptr[0], std::numeric_limits<int32>::max()); in IndicesValidMatrix32BitFastPath()
162 DCHECK_LE(shape_ptr[1], std::numeric_limits<int32>::max()); in IndicesValidMatrix32BitFastPath()
164 const int32 max_rows = static_cast<int32>(shape_ptr[0]); in IndicesValidMatrix32BitFastPath()
165 const int32 max_cols = static_cast<int32>(shape_ptr[1]); in IndicesValidMatrix32BitFastPath()
227 const int64* const shape_ptr = shape_.data(); in IndicesValidHelper() local
235 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_ptr[di]) valid = false; in IndicesValidHelper()
240 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_ptr[di]) valid = false; in IndicesValidHelper()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.h145 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
151 const void* shape_ptr, xla::int32 shape_length);
164 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
177 const void* shape_ptr, xla::int32 shape_length, xla::int32 num_buffers,
Dcpu_runtime.cc189 const void* shape_ptr, xla::int32 size_bytes) { in DecodeSelfDescribingShapeConstant() argument
191 if (!shape_proto.ParseFromArray(shape_ptr, size_bytes)) { in DecodeSelfDescribingShapeConstant()
202 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) { in ShapeString() argument
204 DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in ShapeString()
258 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
263 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
269 DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
277 const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
282 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
294 << ShapeString(shape_ptr, shape_length); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
[all …]
Dir_emitter.cc461 llvm::Value * shape_ptr, in EmitXfeedTransfer()
478 shape_ptr, b_.getInt32(shape_length)}, in EmitXfeedTransfer()
509 acquired_pointer, shape_ptr, b_.getInt32(shape_length)}, in EmitXfeedTransfer()
1171 TF_ASSIGN_OR_RETURN(llvm::Value * shape_ptr, in HandleAllReduceMultipleReplica()
1191 /*shape_ptr=*/shape_ptr, in HandleAllReduceMultipleReplica()