HomeSort by relevance Sort by last modified time
    Searched defs:flat_inner_dims (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/framework/
tensor.h 290 /// auto inner = my_ten.flat_inner_dims<T>();
316 typename TTypes<T, NDIMS>::Tensor flat_inner_dims();
421 typename TTypes<T, NDIMS>::ConstTensor flat_inner_dims() const;
728 typename TTypes<T, NDIMS>::Tensor Tensor::flat_inner_dims() { function in class:tensorflow::Tensor
745 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_inner_dims() const { function in class:tensorflow::Tensor
tensor_test.cc 497 LOG(INFO) << "flat_inner_dims";
499 auto flat_inner_dims = t.flat_inner_dims<float>(); local
500 EXPECT_EQ(24, flat_inner_dims.dimension(0));
501 EXPECT_EQ(5, flat_inner_dims.dimension(1));
502 EXPECT_EQ(flat_inner_dims(0, 0), 0.01f);
503 EXPECT_EQ(flat_inner_dims(23, 4), 0.02f);
506 auto flat_inner_dims = t.flat_inner_dims<float, 3>(); local
507 EXPECT_EQ(6, flat_inner_dims.dimension(0))
514 auto flat_inner_dims = t.flat_inner_dims<float, 5>(); local
524 auto flat_inner_dims = zero_t.flat_inner_dims<float>(); local
529 auto flat_inner_dims = zero_t.flat_inner_dims<float, 3>(); local
535 auto flat_inner_dims = zero_t.flat_inner_dims<float, 5>(); local
    [all...]

Completed in 98 milliseconds