Searched refs:perm_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_test.cc | 179 std::initializer_list<int> perm_shape, in TransposeOpConstModel() argument
182 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeOpConstModel()
200 std::initializer_list<int> perm_shape) { in TransposeOpDynamicModel() argument
206 BuildInterpreter({input_shape, perm_shape}); in TransposeOpDynamicModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 801 std::initializer_list<int> perm_shape, in TransposeSimpleModel() argument
804 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeSimpleModel()
808 BuildInterpreter({input_shape, perm_shape}); in TransposeSimpleModel()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 116 ShapeHandle perm_shape = c->input(1); in TransposeShapeFn() local
118 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn()
148 TF_RETURN_IF_ERROR(c->WithRank(perm_shape, 1, &perm_shape)); in TransposeShapeFn()
|