Searched refs:shape_vector (Results 1 – 3 of 3) sorted by relevance
115 std::vector<int64> shape_vector = std::get<1>(test_params); in TEST_P() local116 TensorShape shape(shape_vector); in TEST_P()126 testing::CreateTensorHandle(context(), dtype, shape_vector, 42); in TEST_P()
237 auto shape_vector = output_type.getShape(); in GetShape() local239 shape.reserve(shape_vector.size()); in GetShape()240 for (auto shape_object : shape_vector) { in GetShape()
208 TensorShape shape_vector({input_channels}); in SimpleFusedBatchNormGrad() local209 Tensor data_vector(DT_FLOAT, shape_vector); in SimpleFusedBatchNormGrad()