Searched refs:dim_t (Results 1 – 4 of 4) sorted by relevance
596 Tensor dim_t; in TEST() local597 op.input_tensors[1] = &dim_t; in TEST()601 dim_t = test::AsScalar<int32>(idx); in TEST()608 dim_t = test::AsScalar<int32>(idx); in TEST()613 dim_t = test::AsScalar<int64>(idx); in TEST()618 dim_t = test::AsScalar<int32>(idx); in TEST()623 dim_t = test::AsScalar<int64>(idx); in TEST()630 dim_t = test::AsScalar<int32>(idx); in TEST()635 dim_t = test::AsScalar<int64>(idx); in TEST()641 dim_t = test::AsScalar<int32>(idx); in TEST()[all …]
923 const Tensor* dim_t = c->input_tensor(1); in ArgOpShape() local924 if (dim_t == nullptr) { in ArgOpShape()938 if (dim_t->dtype() == DT_INT32) { in ArgOpShape()939 dimension_val = dim_t->scalar<int32>()(); in ArgOpShape()941 dimension_val = dim_t->scalar<int64>()(); in ArgOpShape()
1929 const Tensor* dim_t = c->input_tensor(1); in __anondb9326b22f02() local1930 if (dim_t != nullptr && dim_t->NumElements() != 1) { in __anondb9326b22f02()1934 if (dim_t == nullptr || !c->RankKnown(input)) { in __anondb9326b22f02()1940 if (dim_t->dtype() == DT_INT32) { in __anondb9326b22f02()1941 dim = static_cast<int64>(dim_t->flat<int32>()(0)); in __anondb9326b22f02()1943 dim = dim_t->flat<int64>()(0); in __anondb9326b22f02()
1100 struct dim_t1105 struct dim_t<dynamic_range>1109 constexpr dim_t(std::ptrdiff_t size) noexcept : dvalue(size) {}1113 constexpr dim_t<N> dim() noexcept1115 return dim_t<N>();1119 constexpr dim_t<N> dim(std::ptrdiff_t n) noexcept1121 return dim_t<>(n);1191 !std::is_same<Arg, dim_t<dynamic_range>>::value && !std::is_same<Arg, Sep>::value, T>1197 T static_as_multi_span_helper(dim_t<dynamic_range> val, Args... args)1746 constexpr auto as_multi_span(T* const& ptr, dim_t<Dimensions>... args)