Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc276 const int64 primitive_size = in CopyElementFrom() local
280 static_cast<char*>(untyped_data()) + dest_linear_index * primitive_size; in CopyElementFrom()
283 src_linear_index * primitive_size; in CopyElementFrom()
285 memcpy(dest_address, source_address, primitive_size); in CopyElementFrom()
628 const int64 primitive_size = in Broadcast() local
640 memcpy(dest_data + primitive_size * dest_index, in Broadcast()
641 source_data + primitive_size * source_index, primitive_size); in Broadcast()