Searched refs:init_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | hlo-legalize-to-linalg.mlir | 288 // CHECK: linalg.init_tensor [2, 2] : tensor<2x2xi1>
303 // CHECK: linalg.init_tensor [2, 2] : tensor<2x2xi1>
318 // CHECK: linalg.init_tensor [2, 2] : tensor<2x2xi1>
377 // CHECK: linalg.init_tensor [2, 2] : tensor<2x2xf32>
392 // CHECK: linalg.init_tensor [4, 2, 1] : tensor<4x2x1xf32>
408 // CHECK: linalg.init_tensor [4, 2, 1, 4, %[[D1]], 16] : tensor<4x2x1x4x?x16xf32>
424 // CHECK: linalg.init_tensor [7, 10, 6, 4, 5] : tensor<7x10x6x4x5xf32>
441 // CHECK: linalg.init_tensor [1, 5] : tensor<1x5xf32>
457 // CHECK: linalg.init_tensor [7, 10, 6] : tensor<7x10x6xf32>
561 // CHECK: linalg.init_tensor [2, 2] : tensor<2x2xf32>
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | op_selector.py | 368 def map_subgraph(init_tensor, sources, disallowed_placeholders, visited_ops, argument
393 ops_to_visit = [_as_operation(init_tensor)]
413 % (repr(init_tensor), repr(op), _path_from(op, init_tensor, sources)))
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | lift_to_graph.py | 255 for init_tensor in init_tensors:
257 init_tensor=init_tensor,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1146 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite() local
1148 rewriter.create<linalg::FillOp>(loc, init_tensor, zero).getResult(0); in matchAndRewrite()
1234 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite() local
1236 rewriter.create<linalg::FillOp>(loc, init_tensor, zero).getResult(0); in matchAndRewrite()
1353 auto init_tensor = GetInitTensor(rewriter, loc, shaped_type, dyn_shape); in matchAndRewrite() local
1355 rewriter.create<linalg::FillOp>(loc, init_tensor, init_value) in matchAndRewrite()
|
| /external/libaom/libaom/av1/encoder/ |
| D | cnn.c | 48 static void init_tensor(TENSOR *tensor) { memset(tensor, 0, sizeof(*tensor)); } in init_tensor() function
127 init_tensor(&t); in concat_tensor()
850 init_tensor(&tensor1[b]); in av1_cnn_predict_c()
851 init_tensor(&tensor2[b]); in av1_cnn_predict_c()
|