Searched refs:empty_shape (Results 1 – 6 of 6) sorted by relevance
645 TensorShape empty_shape; in Compute() local646 element_shape_.AsTensorShape(&empty_shape); in Compute()647 empty_shape.InsertDim(0, 0); in Compute()649 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, empty_shape, &empty_unused)); in Compute()849 TensorShape empty_shape; in Compute() local850 element_shape_except0_.AsTensorShape(&empty_shape); in Compute()851 empty_shape.InsertDim(0, 0); in Compute()853 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, empty_shape, &empty_unused)); in Compute()
139 TensorShape empty_shape({0}); in TEST_F() local161 auto mean_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()163 auto var_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()246 TensorShape empty_shape({0}); in TEST_F() local273 auto mean_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()275 auto var_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
323 TensorShape empty_shape({0}); in TEST_F() local357 auto mean_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()359 auto var_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
318 TensorShape empty_shape({0}); in TEST() local319 empty_shape.AsProto(proto.mutable_tensor_shape()); in TEST()
476 TensorShape empty_shape({0}); in EmptyConst() local477 empty_shape.AsProto(proto.mutable_tensor_shape()); in EmptyConst()
1784 TensorShape empty_shape; in FastParseSingleExample() local1785 empty_shape.AddDim(0); in FastParseSingleExample()1787 empty_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseSingleExample()1789 result->dense_values[d] = Tensor(config.dense[d].dtype, empty_shape); in FastParseSingleExample()