Home
last modified time | relevance | path

Searched refs:shape_type (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/
Dhvx_ops_support_checker_main.cc93 for (const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type : in DumpRemoteFusedGraph() local
95 LOG(INFO) << "Input shape type: " << shape_type.DebugString(); in DumpRemoteFusedGraph()
100 for (const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type : in DumpRemoteFusedGraph() local
102 LOG(INFO) << "Output shape type: " << shape_type.DebugString(); in DumpRemoteFusedGraph()
/external/tensorflow/tensorflow/python/ops/
Dlist_ops.py188 op.outputs[0], shape_type=dtypes.int32))
204 op.inputs[0], shape_type=dtypes.int32),
236 op.outputs[0], shape_type=dtypes.int32))
253 shape_type=dtypes.int32),
289 input_list, shape_type=dtypes.int32)
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc108 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init() local
110 int64 buf_size = DataTypeSize(shape_type.dtype()); in Init()
111 for (const TensorShapeProto::Dim& dim : shape_type.shape().dim()) { in Init()
119 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init() local
122 int64 buf_size = DataTypeSize(shape_type.dtype()); in Init()
123 for (const TensorShapeProto::Dim& dim : shape_type.shape().dim()) { in Init()
/external/tensorflow/tensorflow/lite/kernels/
Dreshape_test.cc52 ShapeSpecificationType shape_type) { in ReshapeOpModel() argument
53 switch (shape_type) { in ReshapeOpModel()
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_list_ops_test.py41 e32 = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int32)
42 e64 = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int64)
/external/tensorflow/tensorflow/python/kernel_tests/
Dlist_ops_test.py833 l_ps, shape_type=dtypes.int32)
1118 shape = list_ops.tensor_list_element_shape(l, shape_type=dtypes.int32)
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt10257 type_attr: "shape_type"
10272 name: "shape_type"
41424 type_attr: "shape_type"
41443 name: "shape_type"
41461 type_attr: "shape_type"
41464 name: "shape_type"
41482 type_attr: "shape_type"
41493 name: "shape_type"
41625 type_attr: "shape_type"
41640 name: "shape_type"
[all …]
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1929 const tensorflow::DataType shape_type = in ConvertRandomUniformOperator() local
1931 (*new_op->mutable_attr())["T"].set_type(shape_type); in ConvertRandomUniformOperator()
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt21972 type_attr: "shape_type"
21987 name: "shape_type"
84295 type_attr: "shape_type"
84314 name: "shape_type"
84332 type_attr: "shape_type"
84335 name: "shape_type"
84353 type_attr: "shape_type"
84364 name: "shape_type"
84496 type_attr: "shape_type"
84511 name: "shape_type"
[all …]
Dops_history.v1.pbtxt22343 type_attr: "shape_type"
22358 name: "shape_type"
84922 type_attr: "shape_type"
84941 name: "shape_type"
84959 type_attr: "shape_type"
84962 name: "shape_type"
84980 type_attr: "shape_type"
84991 name: "shape_type"
85123 type_attr: "shape_type"
85138 name: "shape_type"
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4037 …argspec: "args=[\'input_handle\', \'shape_type\', \'name\'], varargs=None, keywords=None, defaults…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4037 …argspec: "args=[\'input_handle\', \'shape_type\', \'name\'], varargs=None, keywords=None, defaults…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go34994 func TensorListElementShape(scope *Scope, input_handle tf.Output, shape_type tf.DataType) (element_…
34998 attrs := map[string]interface{}{"shape_type": shape_type}