| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.cc | 53 depth = mklInputShape->GetSizes()[2];
54 tensor_in_cols = mklInputShape->GetSizes()[0];
55 tensor_in_rows = mklInputShape->GetSizes()[1];
56 tensor_in_batch = mklInputShape->GetSizes()[3];
|
| mkl_conv_grad_bias_ops.cc | 78 mkl_context.c_size = mkl_context.input_shape.GetSizes()[MklDims::C];
97 mkl_context.input_shape.GetSizes()[MklDims::W];
99 mkl_context.input_shape.GetSizes()[MklDims::H];
101 mkl_context.input_shape.GetSizes()[MklDims::C];
103 mkl_context.input_shape.GetSizes()[MklDims::N];
|
| mkl_tfconv_op.h | 157 output_shape.AddDim(input_shape.GetSizes()[input_shape.tf_dim_idx(i)]);
158 in_sizes[i] = input_shape.GetSizes()[i];
|
| mkl_concat_op.cc | 341 size_t exp_size = inpshape0.GetSizes()[d];
343 context, exp_size == s.GetSizes()[d],
347 d, "] = ", s.GetSizes()[d]));
356 s.GetDimension() > 0 ? s.GetSizes()[concat_dim] : 1;
378 mkl_context.out_sizes[MklDims::W] = inpshape0.GetSizes()[MklDims::W];
379 mkl_context.out_sizes[MklDims::H] = inpshape0.GetSizes()[MklDims::H];
381 mkl_context.out_sizes[MklDims::N] = inpshape0.GetSizes()[MklDims::N];
[all...] |
| mkl_fused_batch_norm_op.cc | 248 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[0]
251 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[1]
254 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[2]
257 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[3]
531 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[0]
534 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[1]
537 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[2]
540 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[3]
[all...] |
| mkl_input_conversion_op.cc | 194 mkl_shape->GetSizes(),
207 mkl_shape->GetSizes(), mkl_shape->GetStrides());
|
| mkl_reshape_op.cc | 103 mkl_shape_input.GetSizes()[mkl_shape_input.tf_dim_idx(i)]);
|
| mkl_relu_op.cc | 96 mkl_context.in_sizes[i] = mkl_context.input_shape.GetSizes()[i];
306 mkl_context.in_sizes[i] = tmp_mkl_shape->GetSizes()[i];
[all...] |
| mkl_lrn_op.cc | 221 mkl_output_mkl_shape.SetTfLayout(in_dims, input_shape.GetSizes(),
239 mkl_workspace_mkl_shape.SetTfLayout(in_dims, input_shape.GetSizes(),
404 tmp_mkl_shape->GetSizes()[i];
[all...] |
| mkl_aggregate_ops.cc | 114 mkl_context.in_sizes[i] = tmp_mkl_shape->GetSizes()[i];
|
| mkl_conv_grad_input_ops.cc | 112 (const int64*)mkl_context.filter_shape.GetSizes();
|
| /external/tensorflow/tensorflow/core/framework/ |
| tracking_allocator_test.cc | 78 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes();
102 sizes = ta->GetSizes();
127 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes();
152 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes();
170 std::tuple<size_t, size_t, size_t> sizes = ta->GetSizes();
|
| tracking_allocator.h | 82 std::tuple<size_t, size_t, size_t> GetSizes();
|
| tracking_allocator.cc | 161 std::tuple<size_t, size_t, size_t> TrackingAllocator::GetSizes() {
|
| /external/tensorflow/tensorflow/c/eager/ |
| runtime.cc | 307 auto sizes = allocator_pair.second->GetSizes();
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| step_stats_collector.cc | 48 auto sizes = tracking_allocator->GetSizes();
|
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 157 const size_t* GetSizes() const { return sizes_; }
450 inline const int* GetSizes() const {
686 output_shape.AddDim(mkl_shape.GetSizes()[mkl_shape.tf_dim_idx(j)]);
[all...] |
| /development/vndk/tools/header-checker/header-abi-util/include/ |
| ir_representation.h | 1113 const std::pair<uint64_t, uint64_t> &GetSizes() const {
[all...] |
| /development/vndk/tools/header-checker/header-abi-util/src/ |
| ir_representation_protobuf.cpp | 739 const std::pair<uint64_t, uint64_t> &sizes = type_diff_ir->GetSizes();
[all...] |
