/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});
|