Home
last modified time | relevance | path

Searched refs:padding_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dmirror_pad_builder.cc40 const auto& padding_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
41 if (padding_tensor.dims->size != 2 || padding_tensor.dims->data[0] > 4 || in PopulateSubGraph()
42 padding_tensor.dims->data[1] != 2) { in PopulateSubGraph()
50 int padding_data_start = 8 - padding_tensor.dims->data[0] * 2; in PopulateSubGraph()
51 for (int i = 0; i < padding_tensor.dims->data[0] * 2; ++i) { in PopulateSubGraph()
52 padding_data[padding_data_start + i] = padding_tensor.data.i32[i]; in PopulateSubGraph()
H A Dpad_builder.cc41 const auto& padding_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
42 if (padding_tensor.allocation_type == kTfLiteMmapRo) { in PopulateSubGraph()
46 graph_builder_->AddConstNodeWithData(tensor_id, padding_tensor); in PopulateSubGraph()
H A Dtranspose_conv_2d_builder.cc106 std::vector<int> padding_tensor = {padding.height, padding.height, in PopulateSubGraph() local
111 reinterpret_cast<char*>(padding_tensor.data()), (sizeof(int) * 4)); in PopulateSubGraph()
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/AIDL_V3/
Dmirror_pad.mod.py44 padding_tensor = Parameter("padding", ("TENSOR_INT32", [len(input_dims), 2]), paddings)
47 … model = Model().Operation("MIRROR_PAD", input_tensor, padding_tensor, mode).To(output_tensor)
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_flat_param.py676 padding_tensor = _construct_padding_tensor(
679 params_to_flatten.append(padding_tensor)
731 padding_tensor = _construct_padding_tensor(
734 params_to_flatten.append(padding_tensor)
826 padding_tensor = _construct_padding_tensor(
829 flat_tensors.append(padding_tensor)
835 padding_tensor = _construct_padding_tensor(
838 flat_tensors.append(padding_tensor)