HomeSort by relevance Sort by last modified time
    Searched refs:paddings_data (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/
pad.cc 50 const int32* paddings_data = GetTensorData<int32>(paddings); local
54 (paddings_data[0] == 0 && paddings_data[1] == 0) &&
55 (paddings_data[6] == 0 && paddings_data[7] == 0)) {
80 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); local
83 int before_padding = *paddings_data++;
84 int after_padding = *paddings_data++;
138 const int32* paddings_data = GetTensorData<int32>(op_context.paddings); local
145 before_padding.push_back(paddings_data[idx * 2])
    [all...]
space_to_batch_nd.cc 60 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); local
74 int final_dim_size = (input_size->data[dim + 1] + paddings_data[dim * 2] +
75 paddings_data[dim * 2 + 1]);
  /external/tensorflow/tensorflow/tools/graph_transforms/
flatten_atrous_test.cc 55 Tensor paddings_data(DT_INT32, TensorShape({2, 2}));
56 test::FillValues<int>(&paddings_data, {1, 2, 1, 2});
58 Input::Initializer(paddings_data));
  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 1308 const auto& paddings_data = local
    [all...]
  /external/tensorflow/tensorflow/core/ops/
array_ops.cc 61 auto paddings_data = paddings_t->matrix<T>(); local
63 const T pad0 = paddings_data(i, 0);
64 const T pad1 = paddings_data(i, 1);
1809 auto paddings_data = paddings_t->matrix<T>(); local
    [all...]
  /external/tensorflow/tensorflow/lite/kernels/internal/reference/
legacy_reference_ops.h     [all...]
reference_ops.h     [all...]

Completed in 415 milliseconds