Searched refs:zero_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/util/ |
| D | random_seed_test.py | 35 zero_t = constant_op.constant(0, dtype=dtypes.int64, name='zero')
53 ((0, zero_t), (0, 2**31 - 1)), # Avoid nondeterministic (0, 0) output
54 ((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 | 344 Tensor zero_t; member in tensorflow::__anonff72ae010111::TensorReshapeTest
348 zero_t(DT_FLOAT, TensorShape({3, 0, 2, 0, 5})) {}
352 EXPECT_TRUE(zero_t.shape().IsSameSize(TensorShape({3, 0, 2, 0, 5}))); in SetUp()
525 auto flat_inner_dims = zero_t.flat_inner_dims<float>(); in TEST_F()
530 auto flat_inner_dims = zero_t.flat_inner_dims<float, 3>(); in TEST_F()
536 auto flat_inner_dims = zero_t.flat_inner_dims<float, 5>(); in TEST_F()
573 auto flat_outer_dims = zero_t.flat_outer_dims<float>(); in TEST_F()
578 auto flat_outer_dims = zero_t.flat_outer_dims<float, 3>(); in TEST_F()
584 auto flat_outer_dims = zero_t.flat_outer_dims<float, 5>(); in TEST_F()
683 auto flat_inner_outer_dims = zero_t.flat_inner_outer_dims<float, 2>(0); in TEST_F()
[all …]
|