Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_memory_manager.h138 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>> temporary_memory = in AllocateArray() local
140 if (!temporary_memory.ok()) { in AllocateArray()
141 return temporary_memory.status(); in AllocateArray()
146 temporary_memory.ConsumeValueOrDie().release())); in AllocateArray()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h59 Tensor temporary_memory; in AllocateBytes() local
73 DT_UINT8, TensorShape({byte_size}), &temporary_memory, in AllocateBytes()
83 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
86 AsDeviceMemory(temporary_memory.flat<uint8>().data(), in AllocateBytes()
87 temporary_memory.flat<uint8>().size())); in AllocateBytes()
Dfft_ops.cc370 Tensor temporary_memory; in AllocateBytes() local
377 DT_UINT8, TensorShape({byte_size}), &temporary_memory, in AllocateBytes()
384 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
387 AsDeviceMemory(temporary_memory.flat<uint8>().data(), in AllocateBytes()
388 temporary_memory.flat<uint8>().size())); in AllocateBytes()
Dfused_batch_norm_op.cc676 Tensor temporary_memory; in AllocateBytes() local
681 tf_data_type, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
687 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
690 temporary_memory.template flat<T>().data(), in AllocateBytes()
691 temporary_memory.template flat<T>().size() * sizeof(T)); in AllocateBytes()
735 Tensor* temporary_memory = nullptr; in AllocateBytes() local
737 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
743 temporary_memory->template flat<T>().data(), in AllocateBytes()
744 temporary_memory->template flat<T>().size() * sizeof(T)); in AllocateBytes()
Dmatmul_op_impl.h299 Tensor temporary_memory;
302 DT_UINT8, TensorShape({byte_size}), &temporary_memory));
309 allocated_tensors_.push_back(temporary_memory);
312 temporary_memory.flat<uint8>().data(),
313 temporary_memory.flat<uint8>().size()));
Dcudnn_rnn_ops.cc371 Tensor temporary_memory; in AllocateBytes() local
376 tf_data_type, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
382 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
385 temporary_memory.template flat<T>().data(), in AllocateBytes()
386 temporary_memory.template flat<T>().size() * sizeof(T)); in AllocateBytes()
420 Tensor* temporary_memory = nullptr; in AllocateBytes() local
422 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
428 temporary_memory->template flat<T>().data(), in AllocateBytes()
429 temporary_memory->template flat<T>().size() * sizeof(T)); in AllocateBytes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc647 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
672 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
693 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
715 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
738 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
767 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
803 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
815 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
861 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
873 EXPECT_EQ(cost.temporary_memory, 0); in TEST_F()
[all …]
Dcost_estimator.h145 int64 temporary_memory; member
188 temporary_memory = kMemoryUnknown; in Costs()
201 costs.temporary_memory = kZeroMemory; in ZeroCosts()
Danalytical_cost_estimator.cc64 node->set_temporary_memory_size(node_costs.temporary_memory); in AddCostNode()
Dop_level_cost_estimator.h75 int64 temporary_memory = 0; member
Dop_level_cost_estimator.cc679 costs.temporary_memory = node_costs.temporary_memory; in PredictCosts()