Searched refs:zeros_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 84 NodeDef* zeros_shape; in TestSinglePartition() local
97 zeros_shape = CreateNode("w/part_1/Initializer/zeros/shape_as_tensor", in TestSinglePartition()
102 {zeros_shape, zeros_const}, &graph_def); in TestSinglePartition()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 1237 zeros_shape = array_ops.shape_internal(val, optimize=False)
1240 zeros_shape.op._set_control_flow_context(op_ctxt)
1243 zeros_shape = array_ops.shape_internal(val, optimize=False)
1249 zeros_shape, dead_branch=dead_branch)
1254 zeros_shape, dead_branch)
1345 zeros_shape = array_ops.shape_internal(switch_val, optimize=False)
1349 return array_ops.zeros(zeros_shape, dtype=val.dtype)
2698 zeros_shape = array_ops.shape_internal(value, optimize=False)
2702 zeros_shape)
2705 history_zeros_shape, zeros_shape)
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | README.md | 204 …control_flow_ops.py:1313:ZerosLikeOutsideL...:zeros_shape = arr..., cpu: 22.85ms, accelerator: 0us…
|