Searched refs:memory_index (Results 1 – 3 of 3) sorted by relevance
97 tensorflow::int64 memory_index = in __xla_cpu_runtime_KeyValueSort() local102 std::string(values[idx] + memory_index, in __xla_cpu_runtime_KeyValueSort()106 tensorflow::int64 memory_index = (base_offset + i * sort_dimension_offset) * in __xla_cpu_runtime_KeyValueSort() local108 memcpy(values[idx] + memory_index, reordered_values[i].c_str(), in __xla_cpu_runtime_KeyValueSort()
1561 for (uint32_t memory_index = 0; memory_index < memory_snapshots.row_count(); in ExportMemorySnapshots() local1562 ++memory_index) { in ExportMemorySnapshots()1567 auto snapshot_id = memory_snapshots.id()[memory_index].value; in ExportMemorySnapshots()1569 int64_t snapshot_ts = memory_snapshots.timestamp()[memory_index]; in ExportMemorySnapshots()1575 storage_, memory_snapshots.detail_level()[memory_index]); in ExportMemorySnapshots()
465 def memory_index(indices, t): function480 i = memory_index(ind, t)