Home
last modified time | relevance | path

Searched refs:perm_shape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_permutation.py194 perm_shape = self._perm.shape
195 return perm_shape.concatenate(perm_shape[-1:])
198 perm_shape = array_ops.shape(self._perm)
199 k = perm_shape[-1]
200 return array_ops.concat((perm_shape, [k]), 0)
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc251 const TensorShape& perm_shape = permutation_indices.shape(); in ValidateInputs() local
253 ctx, perm_shape.dims() + 1 == rank, in ValidateInputs()
256 " != ", perm_shape.dims(), " + 1.")); in ValidateInputs()
258 ctx, perm_shape.dim_size(rank - 2) == *num_rows, in ValidateInputs()
262 perm_shape.dim_size(rank - 2), " != ", *num_rows)); in ValidateInputs()
267 ctx, perm_shape.dim_size(0) == *batch_size, in ValidateInputs()
270 perm_shape.dim_size(0), " != ", *batch_size)); in ValidateInputs()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_test.cc24 std::initializer_list<int> perm_shape, in TransposeOpModel() argument
29 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeOpModel()
31 perm_ = AddInput({TensorType_INT32, perm_shape}); in TransposeOpModel()
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc584 ShapeHandle perm_shape; in __anon5631e9dc1002() local
585 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &perm_shape)); in __anon5631e9dc1002()
586 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 2, &perm_shape)); in __anon5631e9dc1002()
587 if (!c->RankKnown(perm_shape)) { in __anon5631e9dc1002()
593 TF_RETURN_IF_ERROR(c->Merge(n, c->Dim(perm_shape, -1), &n)); in __anon5631e9dc1002()
599 TF_RETURN_IF_ERROR(c->Subshape(perm_shape, 0, -1, &perm_shape)); in __anon5631e9dc1002()
Darray_ops.cc123 ShapeHandle perm_shape = c->input(1); in TransposeShapeFn() local
125 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn()
155 TF_RETURN_IF_ERROR(c->WithRank(perm_shape, 1, &perm_shape)); in TransposeShapeFn()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_test.cc238 std::initializer_list<int> perm_shape, in TransposeOpConstModel() argument
241 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeOpConstModel()
259 std::initializer_list<int> perm_shape) { in TransposeOpDynamicModel() argument
265 BuildInterpreter({input_shape, perm_shape}); in TransposeOpDynamicModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1547 std::initializer_list<int> perm_shape, in TransposeSimpleModel() argument
1550 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeSimpleModel()
1554 BuildInterpreterWithNNAPI({input_shape, perm_shape}); in TransposeSimpleModel()