Home
last modified time | relevance | path

Searched refs:ShapeInferenceTestOp (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Dtraining_ops_test.cc23 static void TestGradAndIndicesErrorHandling(const ShapeInferenceTestOp& op, in TestGradAndIndicesErrorHandling()
46 ShapeInferenceTestOp op("ApplyGradientDescent"); in TEST()
58 ShapeInferenceTestOp op("ApplyProximalGradientDescent"); in TEST()
72 ShapeInferenceTestOp op("SparseApplyProximalGradientDescent"); in TEST()
86 ShapeInferenceTestOp op("ApplyAdadelta"); in TEST()
106 ShapeInferenceTestOp op("SparseApplyAdadelta"); in TEST()
128 ShapeInferenceTestOp op("ApplyAdagrad"); in TEST()
142 ShapeInferenceTestOp op("SparseApplyAdagrad"); in TEST()
159 ShapeInferenceTestOp op("ApplyProximalAdagrad"); in TEST()
175 ShapeInferenceTestOp op("SparseApplyProximalAdagrad"); in TEST()
[all …]
Dsparse_ops_test.cc26 ShapeInferenceTestOp op("SparseTensorDenseAdd"); in TEST()
33 ShapeInferenceTestOp op("SparseAdd"); in TEST()
43 ShapeInferenceTestOp op("SparseAddGrad"); in TEST()
56 ShapeInferenceTestOp op("SparseSliceGrad"); in TEST()
68 ShapeInferenceTestOp op("SparseReorder"); in TEST()
85 ShapeInferenceTestOp op("SparseReshape"); in TEST()
103 ShapeInferenceTestOp op("SparseSplit"); in TEST()
122 ShapeInferenceTestOp op("SparseToDense"); in TEST()
135 ShapeInferenceTestOp op("SparseReduceSum"); in TEST()
149 ShapeInferenceTestOp op("SerializeSparse"); in TEST()
[all …]
Dsparse_csr_matrix_ops_test.cc29 ShapeInferenceTestOp op("SparseTensorToCSRSparseMatrix"); in TEST()
44 ShapeInferenceTestOp op("CSRSparseMatrixToSparseTensor"); in TEST()
45 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
60 ShapeInferenceTestOp op("DenseToCSRSparseMatrix"); in TEST()
71 ShapeInferenceTestOp op("CSRSparseMatrixToDense"); in TEST()
72 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
83 ShapeInferenceTestOp op("CSRSparseMatrixComponents"); in TEST()
84 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
103 ShapeInferenceTestOp op("SparseMatrixMatMul"); in TEST()
104 std::vector<ShapeInferenceTestOp::ShapeAndType> a_shapes_and_types(1); in TEST()
[all …]
Dio_ops_test.cc25 ShapeInferenceTestOp op("Save"); in TEST()
45 ShapeInferenceTestOp op("SaveSlices"); in TEST()
72 ShapeInferenceTestOp op("Restore"); in TEST()
83 ShapeInferenceTestOp op("RestoreV2"); in TEST()
103 ShapeInferenceTestOp op("RestoreSlice"); in TEST()
115 ShapeInferenceTestOp op("ShardedFilename"); in TEST()
127 ShapeInferenceTestOp op("ShardedFilespec"); in TEST()
139 ShapeInferenceTestOp op(op_name); in TEST()
151 ShapeInferenceTestOp op(op_name); in TEST()
161 ShapeInferenceTestOp op("ReaderRead"); in TEST()
[all …]
Dnn_ops_test.cc27 ShapeInferenceTestOp op("TopK"); in TEST()
56 ShapeInferenceTestOp op("TopKV2"); in TEST()
85 ShapeInferenceTestOp op("NthElement"); in TEST()
108 ShapeInferenceTestOp op("BatchNormWithGlobalNormalization"); in TEST()
131 ShapeInferenceTestOp op("QuantizedBatchNormWithGlobalNormalization"); in TEST()
160 ShapeInferenceTestOp op("BatchNormWithGlobalNormalizationGrad"); in TEST()
182 ShapeInferenceTestOp op("FusedBatchNorm"); in TEST()
278 ShapeInferenceTestOp op("FusedBatchNormGrad"); in TEST()
324 ShapeInferenceTestOp op("Conv2DBackpropInput"); in TEST()
343 ShapeInferenceTestOp op("Conv3DBackpropInput"); in TEST()
[all …]
Darray_ops_test.cc31 ShapeInferenceTestOp op("TensorScatterUpdate"); in TEST()
52 ShapeInferenceTestOp op("ScatterNd"); in TEST()
64 ShapeInferenceTestOp op("UnravelIndex"); in TEST()
79 ShapeInferenceTestOp op("Pack"); in TEST()
131 ShapeInferenceTestOp op("Unpack"); in TEST()
170 ShapeInferenceTestOp op("Const"); in TEST()
201 ShapeInferenceTestOp op(op_name); in TEST()
208 ShapeInferenceTestOp op("MatrixBandPart"); in TEST()
215 ShapeInferenceTestOp op("GuaranteeConst"); in TEST()
223 ShapeInferenceTestOp op(op_name); in TEST()
[all …]
Ddata_flow_ops_test.cc27 ShapeInferenceTestOp op("LookupTableFind"); in TEST()
38 ShapeInferenceTestOp op("LookupTableInsert"); in TEST()
48 ShapeInferenceTestOp op("LookupTableSize"); in TEST()
59 ShapeInferenceTestOp op("LookupTableExport"); in TEST()
69 ShapeInferenceTestOp op("InitializeTable"); in TEST()
83 ShapeInferenceTestOp op("InitializeTableFromTextFile"); in TEST()
93 ShapeInferenceTestOp op("DynamicPartition"); in TEST()
121 ShapeInferenceTestOp op("DynamicStitch"); in TEST()
165 ShapeInferenceTestOp op("ParallelDynamicStitch"); in TEST()
185 ShapeInferenceTestOp op("TensorArrayV3"); in TEST()
[all …]
Dimage_ops_test.cc27 ShapeInferenceTestOp op("SampleDistortedBoundingBox"); in TEST()
34 ShapeInferenceTestOp op(op_name); in TEST()
55 ShapeInferenceTestOp op("DecodeGif"); in TEST()
66 ShapeInferenceTestOp op("DecodeImage"); in TEST()
95 ShapeInferenceTestOp op(op_name); in TEST()
124 ShapeInferenceTestOp op(op_name); in TEST()
158 ShapeInferenceTestOp op(op_name); in TEST()
176 ShapeInferenceTestOp op(op_name); in TEST()
186 ShapeInferenceTestOp op("ExtractJpegShape"); in TEST()
197 ShapeInferenceTestOp op(op_name); in TEST()
[all …]
Dmath_ops_test.cc29 ShapeInferenceTestOp op("AddN"); in TEST()
78 ShapeInferenceTestOp op("Cast"); in TEST()
88 ShapeInferenceTestOp op(op_name); in TEST()
113 auto test_shapes = [&](ShapeInferenceTestOp& op, in TEST()
175 ShapeInferenceTestOp op(op_name); in TEST()
180 ShapeInferenceTestOp op(op_name); in TEST()
188 ShapeInferenceTestOp op("Select"); in TEST()
285 ShapeInferenceTestOp op("Range"); in TEST()
336 ShapeInferenceTestOp op("LinSpace"); in TEST()
356 ShapeInferenceTestOp op("UnsortedSegmentSum"); in TEST()
[all …]
Dstate_ops_test.cc25 ShapeInferenceTestOp op("Assign"); in TEST()
51 ShapeInferenceTestOp op("ScatterUpdate"); in TEST()
73 ShapeInferenceTestOp op("ResourceScatterNdUpdate"); in TEST()
80 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; in TEST()
95 ShapeInferenceTestOp op("TemporaryVariable"); in TEST()
104 ShapeInferenceTestOp op("Variable"); in TEST()
127 ShapeInferenceTestOp op("VariableV2"); in TEST()
Dlinalg_ops_test.cc25 ShapeInferenceTestOp op("MatrixDeterminant"); in TEST()
42 ShapeInferenceTestOp op(op_name); in TEST()
65 ShapeInferenceTestOp op("SelfAdjointEig"); in TEST()
81 ShapeInferenceTestOp op("SelfAdjointEigV2"); in TEST()
126 ShapeInferenceTestOp op("MatrixSolve"); in TEST()
153 ShapeInferenceTestOp op("MatrixTriangularSolve"); in TEST()
176 ShapeInferenceTestOp op("MatrixSolveLs"); in TEST()
201 ShapeInferenceTestOp op("Qr"); in TEST()
252 ShapeInferenceTestOp op("Svd"); in TEST()
319 ShapeInferenceTestOp op("Lu"); in TEST()
[all …]
Dset_ops_test.cc25 ShapeInferenceTestOp op("DenseToDenseSetOperation"); in TEST()
31 ShapeInferenceTestOp op("DenseToDenseSetOperation"); in TEST()
81 ShapeInferenceTestOp op("DenseToSparseSetOperation"); in TEST()
87 ShapeInferenceTestOp op("DenseToSparseSetOperation"); in TEST()
126 ShapeInferenceTestOp op("SparseToSparseSetOperation"); in TEST()
132 ShapeInferenceTestOp op("SparseToSparseSetOperation"); in TEST()
Drnn_ops_test.cc35 ShapeInferenceTestOp op("GRUBlockCell"); in TEST()
48 ShapeInferenceTestOp op("GRUBlockCellGrad"); in TEST()
62 ShapeInferenceTestOp op("LSTMBlockCell"); in TEST()
77 ShapeInferenceTestOp op("LSTMBlockCellGrad"); in TEST()
95 ShapeInferenceTestOp op("BlockLSTM"); in TEST()
127 ShapeInferenceTestOp op("BlockLSTMGrad"); in TEST()
Dspectral_ops_test.cc24 ShapeInferenceTestOp op(op_name); in TEST()
33 ShapeInferenceTestOp op(op_name); in TEST()
42 ShapeInferenceTestOp op(op_name); in TEST()
54 ShapeInferenceTestOp op(forward ? "RFFT" : "IRFFT"); in TEST()
114 ShapeInferenceTestOp op(forward ? "RFFT2D" : "IRFFT2D"); in TEST()
174 ShapeInferenceTestOp op(forward ? "RFFT3D" : "IRFFT3D"); in TEST()
Dlookup_ops_test.cc26 ShapeInferenceTestOp op("LookupTableFindV2"); in TEST()
35 std::vector<std::vector<ShapeInferenceTestOp::ShapeAndType>> types; in TEST()
59 ShapeInferenceTestOp op("LookupTableExportV2"); in TEST()
65 std::vector<std::vector<ShapeInferenceTestOp::ShapeAndType>> types; in TEST()
Dctc_ops_test.cc25 ShapeInferenceTestOp op("CTCLoss"); in TEST()
49 ShapeInferenceTestOp op("CTCGreedyDecoder"); in TEST()
67 ShapeInferenceTestOp op("CTCBeamSearchDecoder"); in TEST()
Drandom_ops_test.cc25 ShapeInferenceTestOp op("Multinomial"); in TEST()
41 ShapeInferenceTestOp op("RandomGamma"); in TEST()
56 ShapeInferenceTestOp op("RandomPoisson"); in TEST()
Dcudnn_rnn_ops_test.cc28 ShapeInferenceTestOp op("CudnnRNNParamsSize"); in TEST()
60 ShapeInferenceTestOp op("CudnnRNN"); in TEST()
97 ShapeInferenceTestOp op("CudnnRNNV2"); in TEST()
138 ShapeInferenceTestOp op("CudnnRNNV3"); in TEST()
180 ShapeInferenceTestOp op("CudnnRNNV3"); in TEST()
Dparsing_ops_test.cc26 ShapeInferenceTestOp op("DecodeRaw"); in TEST()
34 ShapeInferenceTestOp op("DecodeCSV"); in TEST()
87 ShapeInferenceTestOp op("ParseExample"); in TEST()
153 ShapeInferenceTestOp op("ParseSequenceExample"); in TEST()
235 ShapeInferenceTestOp op("ParseSingleSequenceExample"); in TEST()
304 ShapeInferenceTestOp op("ParseExampleV2"); in TEST()
390 ShapeInferenceTestOp op("ParseSequenceExampleV2"); in TEST()
Dcontrol_flow_ops_test.cc26 ShapeInferenceTestOp op("Merge"); in TEST()
53 ShapeInferenceTestOp op("_SwitchN"); in TEST()
74 ShapeInferenceTestOp op("RefSelect"); in TEST()
Dtpu_cross_replica_ops_test.cc25 ShapeInferenceTestOp op("AllToAll"); in TEST()
32 ShapeInferenceTestOp op("AllToAll"); in TEST()
Dcandidate_sampling_ops_test.cc33 ShapeInferenceTestOp op(op_name); in TEST()
53 ShapeInferenceTestOp op("ComputeAccidentalHits"); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dcount_ops_test.cc31 ShapeInferenceTestOp op("DenseCountSparseOutput"); in TEST_F()
37 ShapeInferenceTestOp op("SparseCountSparseOutput"); in TEST_F()
43 ShapeInferenceTestOp op("RaggedCountSparseOutput"); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_testutil.h34 struct ShapeInferenceTestOp { struct
36 explicit ShapeInferenceTestOp(StringPiece name) : name(string(name)) {} in ShapeInferenceTestOp() argument
70 static Status InferShapes(ShapeInferenceTestOp op, const string& ins,
Dcommon_shape_fns_test.cc219 ShapeInferenceTestOp op("Einsum"); in TEST()
342 ShapeInferenceTestOp op("BatchMatMulV2"); in TEST()
513 ShapeInferenceTestOp op("_FusedBatchNormEx"); in TEST()
628 ShapeInferenceTestOp op("Conv2D"); in TEST()
820 ShapeInferenceTestOp op("Conv2D"); in TEST()
936 ShapeInferenceTestOp op("Conv3D"); in TEST()
1037 ShapeInferenceTestOp op("Conv3D"); in TEST()
1080 ShapeInferenceTestOp op("DepthwiseConv2dNative"); in TEST()
1124 ShapeInferenceTestOp op("AvgPool"); in TEST()
1170 ShapeInferenceTestOp op("MaxPool"); in TEST()
[all …]

12