Searched refs:GetSizes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tracking_allocator_test.cc | 78 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes(); in TEST()
102 sizes = ta->GetSizes(); in TEST()
127 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes(); in TEST()
152 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes(); in TEST()
170 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes(); in TEST()
|
| D | tracking_allocator.h | 81 std::tuple<size_t, size_t, size_t> GetSizes();
|
| D | tracking_allocator.cc | 161 std::tuple<size_t, size_t, size_t> TrackingAllocator::GetSizes() { in GetSizes() function in tensorflow::TrackingAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_grad_bias_ops.cc | 81 mkl_context.c_size = mkl_context.input_shape.GetSizes()[MklDims::C]; in Compute()
101 mkl_context.input_shape.GetSizes()[MklDims::W]; in Compute()
103 mkl_context.input_shape.GetSizes()[MklDims::H]; in Compute()
105 mkl_context.input_shape.GetSizes()[MklDims::C]; in Compute()
107 mkl_context.input_shape.GetSizes()[MklDims::N]; in Compute()
|
| D | mkl_pooling_ops_common.cc | 258 depth = mklInputShape->GetSizes()[2]; in Init()
259 tensor_in_cols = mklInputShape->GetSizes()[0]; in Init()
260 tensor_in_rows = mklInputShape->GetSizes()[1]; in Init()
261 tensor_in_batch = mklInputShape->GetSizes()[3]; in Init()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 183 auto sizes = allocator_pair.second->GetSizes(); in UpdateStats()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 242 const size_t* GetSizes() const { return sizes_; } in GetSizes() function
561 inline const int* GetSizes() const { in GetSizes() function
810 output_shape.AddDim(mkl_shape.GetSizes()[mkl_shape.tf_dim_idx(j)]); in ConvertMklToTF()
1093 const size_t* tf_sizes = mkl_shape.GetSizes(); in MklSizesToTFSizes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.cc | 174 auto sizes = tracking_allocator->GetSizes(); in AddAllocation()
|