Searched refs:shape_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform.py | 161 shape_op = _find_op(graph_def, shape)
163 shape_op_value_tensor = shape_op.attr['value'].tensor
183 if not shape_op.attr['_output_shapes'].list.shape:
184 shape_op.attr['_output_shapes'].list.shape.add()
185 shape_op.attr['_output_shapes'].list.shape[0].dim.add()
186 shape_op.attr['_output_shapes'].list.shape[0].dim[0].size = len(shapes)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_test.py | 848 shape_op = lambda x: array_ops.shape_internal(x, optimize=False) function
851 shape_op, dtype(7), expected=np.array([], dtype=np.int32))
853 shape_op,
857 shape_op,
861 shape_op,
865 shape_op,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 394 Output shape_op = in TestFoldFusedBatchNormsWithConcat() local
397 Concat(root.WithOpName("concat_op"), {conv0_op, conv1_op}, shape_op); in TestFoldFusedBatchNormsWithConcat()
|
| D | quantize_nodes_test.cc | 373 Output shape_op = in TestQuantizeConcat() local
386 Concat(root.WithOpName("concat_op"), {a_op, b_op}, shape_op); in TestQuantizeConcat()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1297 tensorflow::NodeDef* shape_op = tensorflow_graph->add_node(); in ConvertTensorFlowShapeOperator() local
1298 shape_op->set_op("Shape"); in ConvertTensorFlowShapeOperator()
1299 shape_op->set_name(src_op.outputs[0]); in ConvertTensorFlowShapeOperator()
1301 *shape_op->add_input() = src_op.inputs[0]; in ConvertTensorFlowShapeOperator()
1302 (*shape_op->mutable_attr())["T"].set_type( in ConvertTensorFlowShapeOperator()
1304 (*shape_op->mutable_attr())["out_type"].set_type( in ConvertTensorFlowShapeOperator()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 842 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
843 TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get()); in TEST()
847 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
862 TFE_DeleteOp(shape_op); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 84 "shape_op.cc",
|