Searched refs:concat_dim_t (Results 1 – 2 of 2) sorted by relevance
725 Tensor concat_dim_t; in TEST() local726 op.input_tensors.push_back(&concat_dim_t); in TEST()731 concat_dim_t = test::AsScalar(concat_dim); in TEST()742 concat_dim_t = test::AsScalar(use_negative ? -2 : 1); in TEST()744 concat_dim_t = test::AsScalar(use_negative ? -1 : 1); in TEST()749 concat_dim_t = test::AsScalar(use_negative ? -2 : 1); in TEST()757 concat_dim_t = test::AsScalar(-2); in TEST()765 concat_dim_t = test::AsScalar(1); in TEST()799 Tensor concat_dim_t; in TEST() local801 op.input_tensors[2] = &concat_dim_t; in TEST()[all …]
1215 const Tensor* concat_dim_t = c->input_tensor(dim_index); in ConcatShapeHelper() local1216 if (concat_dim_t == nullptr) { in ConcatShapeHelper()1250 const int32 concat_dim = concat_dim_t->scalar<int32>()(); in ConcatShapeHelper()