Searched refs:TF_ANNOTATE_MEMORY_IS_INITIALIZED (Results 1 – 12 of 12) 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 | 35 TF_ANNOTATE_MEMORY_IS_INITIALIZED(values, values_count * sizeof(char*)); in __xla_cpu_runtime_KeyValueSort()
36 TF_ANNOTATE_MEMORY_IS_INITIALIZED(values_primitive_type_size_in_bytes, in __xla_cpu_runtime_KeyValueSort()
|
| D | runtime_topk.cc | 32 TF_ANNOTATE_MEMORY_IS_INITIALIZED(values, in TopK()
|
| D | cpu_executable.cc | 123 TF_ANNOTATE_MEMORY_IS_INITIALIZED(out->opaque(), buffer_size); in MemoryForAllocation()
|
| /external/tensorflow/tensorflow/core/platform/default/ |
| D | dynamic_annotations.h | 24 #define TF_ANNOTATE_MEMORY_IS_INITIALIZED(ptr, bytes) \ macro
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | cpu_function_runtime.cc | 84 TF_ANNOTATE_MEMORY_IS_INITIALIZED(contiguous, total); in MallocContiguousBuffers()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator.cc | 118 TF_ANNOTATE_MEMORY_IS_INITIALIZED(extra_buf, extra_bytes); in AllocateRaw()
|
| /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 | eigen_contraction_kernel.h | 182 TF_ANNOTATE_MEMORY_IS_INITIALIZED(row_base, sizeof(ResScalar) * rows);
259 TF_ANNOTATE_MEMORY_IS_INITIALIZED(row_base, sizeof(ResScalar) * rows);
|
| 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()
|