HomeSort by relevance Sort by last modified time
    Searched refs:paddings (Results 1 - 25 of 129) sorted by null

1 2 3 4 5 6

  /external/tensorflow/tensorflow/compiler/tests/
spacetobatch_op_test.py 29 def space_to_batch_direct(input_array, block_shape, paddings):
37 paddings: 2-D array of shape [num_block_dims, 2].
45 paddings = np.array(paddings).reshape((len(block_shape), 2))
48 pad_width=([[0, 0]] + list(paddings) + [[0, 0]] *
73 def _testPad(self, inputs, paddings, block_size, outputs):
79 placeholder, paddings, block_size=block_size)
83 placeholder, paddings, block_size=block_size)
87 paddings = np.zeros((2, 2), dtype=np.int32)
88 self._testPad(inputs, paddings, block_size, outputs
    [all...]
  /frameworks/ml/nn/runtime/test/specs/V1_1/
space_to_batch.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [0, 0, 0, 0]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_float_1.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [0, 0, 0, 0]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_float_1_relaxed.mod.py 20 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [0, 0, 0, 0]) variable
23 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_float_2.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [1, 0, 2, 0]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_float_2_relaxed.mod.py 20 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [1, 0, 2, 0]) variable
23 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_float_3.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [1, 1, 2, 4]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_float_3_relaxed.mod.py 20 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [1, 1, 2, 4]) variable
23 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_quant8_1.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [0, 0, 0, 0]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_quant8_2.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [1, 0, 2, 0]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_quant8_3.mod.py 4 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [1, 1, 2, 4]) variable
7 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
space_to_batch_relaxed.mod.py 20 paddings = Parameter("paddings", "TENSOR_INT32", "{2, 2}", [0, 0, 0, 0]) variable
23 model = model.Operation("SPACE_TO_BATCH_ND", i1, block, paddings).To(output)
  /external/eigen/unsupported/test/
cxx11_tensor_padding.cpp 22 array<std::pair<ptrdiff_t, ptrdiff_t>, 4> paddings; local
23 paddings[0] = std::make_pair(0, 0);
24 paddings[1] = std::make_pair(2, 1);
25 paddings[2] = std::make_pair(3, 4);
26 paddings[3] = std::make_pair(0, 0);
29 padded = tensor.pad(paddings);
57 array<std::pair<ptrdiff_t, ptrdiff_t>, 4> paddings; local
58 paddings[0] = std::make_pair(0, 0);
59 paddings[1] = std::make_pair(2, 1);
60 paddings[2] = std::make_pair(3, 4)
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
spacetobatch_op_test.py 33 def space_to_batch_direct(input_array, block_shape, paddings):
41 paddings: 2-D array of shape [num_block_dims, 2].
49 paddings = np.array(paddings).reshape((len(block_shape), 2))
52 pad_width=([[0, 0]] + list(paddings) + [[0, 0]] *
102 def _testPad(self, inputs, paddings, block_size, outputs):
106 math_ops.to_float(inputs), paddings, block_size=block_size)
110 math_ops.to_float(outputs), paddings, block_size=block_size)
114 paddings = np.zeros((2, 2), dtype=np.int32)
115 self._testPad(inputs, paddings, block_size, outputs
    [all...]
  /frameworks/ml/nn/runtime/test/generated/models/
space_to_batch.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_float_1.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_float_1_relaxed.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_float_2.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_float_2_relaxed.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_float_3.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_float_3_relaxed.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_quant8_1.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_quant8_2.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_quant8_3.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});
space_to_batch_relaxed.model.cpp 10 auto paddings = model->addOperand(&type2); local
16 model->setOperandValue(paddings, paddings_init, sizeof(int32_t) * 4);
17 model->addOperation(ANEURALNETWORKS_SPACE_TO_BATCH_ND, {input, block_size, paddings}, {output});

Completed in 340 milliseconds

1 2 3 4 5 6