Searched refs:TF_ANNOTATE_MEMORY_IS_INITIALIZED (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 39 TF_ANNOTATE_MEMORY_IS_INITIALIZED(in, sizeof(float*)); in R0F32Add2()
44 TF_ANNOTATE_MEMORY_IS_INITIALIZED(in, sizeof(float) * 4); in R2F32ReduceSum()
50 TF_ANNOTATE_MEMORY_IS_INITIALIZED(in, sizeof(float) * 4); in Add1ToValues()
59 TF_ANNOTATE_MEMORY_IS_INITIALIZED(in[0], sizeof(float)); in F32TupleSwap()
60 TF_ANNOTATE_MEMORY_IS_INITIALIZED(in[1], sizeof(float)); in F32TupleSwap()
|
| D | local_client_aot_test.cc | 37 TF_ANNOTATE_MEMORY_IS_INITIALIZED(parameters, sizeof(OpaqueData*)); in SumStructElements()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_key_value_sort.cc | 40 TF_ANNOTATE_MEMORY_IS_INITIALIZED(values, values_count * sizeof(char*)); in __xla_cpu_runtime_KeyValueSort()
41 TF_ANNOTATE_MEMORY_IS_INITIALIZED(values_primitive_type_size_in_bytes, in __xla_cpu_runtime_KeyValueSort()
|
| D | cpu_executable.cc | 126 TF_ANNOTATE_MEMORY_IS_INITIALIZED(owning_buffers[i].opaque(), buffer_size); in CreateBufferTable()
|
| /external/tensorflow/tensorflow/core/platform/default/ |
| D | dynamic_annotations.h | 24 #define TF_ANNOTATE_MEMORY_IS_INITIALIZED(ptr, bytes) \ macro
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops_kernel_argmax_float_2d.cc | 29 TF_ANNOTATE_MEMORY_IS_INITIALIZED(data, 4 * sizeof(void*)); in argmax_float_2d_xla_impl()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | cpu_function_runtime.cc | 84 TF_ANNOTATE_MEMORY_IS_INITIALIZED(contiguous, total); in MallocContiguousBuffers()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_matmul_op.cc | 63 TF_ANNOTATE_MEMORY_IS_INITIALIZED(c_data_as_int32, m * n * sizeof(int32)); in GemmlowpMultiply()
|
| D | quantized_conv_ops.cc | 436 TF_ANNOTATE_MEMORY_IS_INITIALIZED(output_data_as_int32, in operator ()()
|
| /external/tensorflow/tensorflow/core/lib/jpeg/ |
| D | jpeg_mem.cc | 349 TF_ANNOTATE_MEMORY_IS_INITIALIZED(output_line, min_stride); in UncompressLow()
|