Searched refs:concat_dim_t (Results 1 – 2 of 2) sorted by relevance
756 Tensor concat_dim_t; in TEST() local757 op.input_tensors.push_back(&concat_dim_t); in TEST()762 concat_dim_t = test::AsScalar(concat_dim); in TEST()773 concat_dim_t = test::AsScalar(use_negative ? -2 : 1); in TEST()775 concat_dim_t = test::AsScalar(use_negative ? -1 : 1); in TEST()780 concat_dim_t = test::AsScalar(use_negative ? -2 : 1); in TEST()788 concat_dim_t = test::AsScalar(-2); in TEST()796 concat_dim_t = test::AsScalar(1); in TEST()830 Tensor concat_dim_t; in TEST() local832 op.input_tensors[2] = &concat_dim_t; in TEST()[all …]
1889 const Tensor* concat_dim_t = c->input_tensor(dim_index); in ConcatShapeHelper() local1890 if (concat_dim_t == nullptr) { in ConcatShapeHelper()1925 if (concat_dim_t->dtype() == DT_INT32) { in ConcatShapeHelper()1926 concat_dim = static_cast<int64>(concat_dim_t->flat<int32>()(0)); in ConcatShapeHelper()1928 concat_dim = concat_dim_t->flat<int64>()(0); in ConcatShapeHelper()