HomeSort by relevance Sort by last modified time
    Searched refs:perm_shape (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/
transpose_test.cc 173 // SpaceToBatchNDOpConstModel m(input_shape, perm_shape, perm_data);
179 std::initializer_list<int> perm_shape,
182 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape);
193 // TransposeOpDynamicModel m(input_shape, perm_shape);
200 std::initializer_list<int> perm_shape) {
206 BuildInterpreter({input_shape, perm_shape});
  /external/tensorflow/tensorflow/lite/delegates/nnapi/
nnapi_delegate_test.cc 801 std::initializer_list<int> perm_shape,
804 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape);
808 BuildInterpreter({input_shape, perm_shape});
    [all...]
  /external/tensorflow/tensorflow/core/ops/
array_ops.cc 116 ShapeHandle perm_shape = c->input(1); local
118 DimensionHandle perm_elems = c->NumElements(perm_shape);
148 TF_RETURN_IF_ERROR(c->WithRank(perm_shape, 1, &perm_shape));
    [all...]

Completed in 182 milliseconds