HomeSort by relevance Sort by last modified time
    Searched refs:shape_op (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/meta_graph_transform/
meta_graph_transform.py 157 shape_op = _find_op(graph_def, shape)
159 shape_op_value_tensor = shape_op.attr['value'].tensor
179 if not shape_op.attr['_output_shapes'].list.shape:
180 shape_op.attr['_output_shapes'].list.shape.add()
181 shape_op.attr['_output_shapes'].list.shape[0].dim.add()
182 shape_op.attr['_output_shapes'].list.shape[0].dim[0].size = len(shapes)
  /external/tensorflow/tensorflow/compiler/tests/
unary_ops_test.py 629 shape_op = lambda x: array_ops.shape_internal(x, optimize=False)
632 shape_op, dtype(7), expected=np.array([], dtype=np.int32))
634 shape_op,
638 shape_op,
642 shape_op,
646 shape_op,
  /external/tensorflow/tensorflow/tools/graph_transforms/
fold_old_batch_norms_test.cc 238 Output shape_op = local
239 Const(root.WithOpName("shape_op"), Input::Initializer(shape_tensor));
241 Concat(root.WithOpName("concat_op"), {conv0_op, conv1_op}, shape_op);
quantize_nodes_test.cc 373 Output shape_op = local
374 Const(root.WithOpName("shape_op"), Input::Initializer(shape_tensor));
386 Concat(root.WithOpName("concat_op"), {a_op, b_op}, shape_op);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 1084 auto* shape_op = tensorflow_graph->add_node(); local
    [all...]

Completed in 588 milliseconds