Searched refs:ScatterGatherShapeInferenceTest (Results 1 – 1 of 1) sorted by relevance
1753 class ScatterGatherShapeInferenceTest : public ShapeInferenceTest { class1776 TEST_F(ScatterGatherShapeInferenceTest, TensorFlowGather) { in TEST_F() argument1791 TEST_F(ScatterGatherShapeInferenceTest, TensorFlowGatherV2) { in TEST_F() argument1806 TEST_F(ScatterGatherShapeInferenceTest, TensorFlowGatherNd) { in TEST_F() argument1821 TEST_F(ScatterGatherShapeInferenceTest, TensorFlowBatchDynamicSlice) { in TEST_F() argument1838 TEST_F(ScatterGatherShapeInferenceTest, NonDefaultGatherIndicesLeafDim_A) { in TEST_F() argument1856 TEST_F(ScatterGatherShapeInferenceTest, NonDefaultGatherIndicesLeafDim_B) { in TEST_F() argument1874 TEST_F(ScatterGatherShapeInferenceTest, NoOutputGatherDims) { in TEST_F() argument1891 TEST_F(ScatterGatherShapeInferenceTest, ScalarGatherIndices) { in TEST_F() argument1909 TEST_F(ScatterGatherShapeInferenceTest, TupleShapedTensorInput) { in TEST_F() argument[all …]