Searched refs:zero_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/util/ |
| D | random_seed_test.py | 34 zero_t = constant_op.constant(0, dtype=dtypes.int64, name='zero')
52 ((0, zero_t), (0, 2**31 - 1)), # Avoid nondeterministic (0, 0) output
53 ((2**31 - 1, zero_t), (0, 2**31 - 1)), # Don't wrap to (0, 0) either
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.h | 116 Tensor zero_t; in OptionalZerosLike() local
117 TF_RETURN_IF_ERROR(ZerosLikeTensor<Device>(ctx, tensor, &zero_t)); in OptionalZerosLike()
118 zero_tensors.push_back(std::move(zero_t)); in OptionalZerosLike()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 374 Tensor zero_t; member in tensorflow::__anon92b39c9f0111::TensorReshapeTest
378 zero_t(DT_FLOAT, TensorShape({3, 0, 2, 0, 5})) {}
382 EXPECT_TRUE(zero_t.shape().IsSameSize(TensorShape({3, 0, 2, 0, 5}))); in SetUp()
558 auto flat_inner_dims = zero_t.flat_inner_dims<float>(); in TEST_F()
563 auto flat_inner_dims = zero_t.flat_inner_dims<float, 3>(); in TEST_F()
569 auto flat_inner_dims = zero_t.flat_inner_dims<float, 5>(); in TEST_F()
606 auto flat_outer_dims = zero_t.flat_outer_dims<float>(); in TEST_F()
611 auto flat_outer_dims = zero_t.flat_outer_dims<float, 3>(); in TEST_F()
617 auto flat_outer_dims = zero_t.flat_outer_dims<float, 5>(); in TEST_F()
716 auto flat_inner_outer_dims = zero_t.flat_inner_outer_dims<float, 2>(0); in TEST_F()
[all …]
|