Home
last modified time | relevance | path

Searched refs:flat_inner_outer_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc596 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 4>(0); in TEST_F() local
597 EXPECT_EQ(2, flat_inner_outer_dims.dimension(0)); in TEST_F()
598 EXPECT_EQ(3, flat_inner_outer_dims.dimension(1)); in TEST_F()
599 EXPECT_EQ(4, flat_inner_outer_dims.dimension(2)); in TEST_F()
600 EXPECT_EQ(5, flat_inner_outer_dims.dimension(3)); in TEST_F()
601 EXPECT_EQ(flat_inner_outer_dims(0, 0, 0, 0), 0.01f); in TEST_F()
602 EXPECT_EQ(flat_inner_outer_dims(1, 2, 3, 4), 0.02f); in TEST_F()
605 auto flat_inner_outer_dims = t.flat_inner_outer_dims<float, 6>(-2); in TEST_F() local
606 EXPECT_EQ(1, flat_inner_outer_dims.dimension(0)); in TEST_F()
607 EXPECT_EQ(1, flat_inner_outer_dims.dimension(1)); in TEST_F()
[all …]
Dtensor.h427 typename TTypes<T, NDIMS>::Tensor flat_inner_outer_dims(int64 begin);
522 typename TTypes<T, NDIMS>::ConstTensor flat_inner_outer_dims(
889 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_outer_dims(int64 begin) { in flat_inner_outer_dims() function
906 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_outer_dims( in flat_inner_outer_dims() function
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/
Dreduce_slice_ops.cc152 indices.flat<Index>(), data.flat_inner_outer_dims<T, 3>(axis - 1), in Compute()
153 output->flat_inner_outer_dims<T, 3>(axis - 1)); in Compute()