Home
last modified time | relevance | path

Searched refs:ArraySlice (Results 1 – 25 of 195) sorted by relevance

12345678

/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.h70 void SetAttrValue(gtl::ArraySlice<string> value, AttrValue* out);
71 void SetAttrValue(gtl::ArraySlice<const char*> value, AttrValue* out);
72 void SetAttrValue(gtl::ArraySlice<StringPiece> value, AttrValue* out);
73 void SetAttrValue(gtl::ArraySlice<int64> value, AttrValue* out);
74 void SetAttrValue(gtl::ArraySlice<int32> value, AttrValue* out);
75 void SetAttrValue(gtl::ArraySlice<float> value, AttrValue* out);
76 void SetAttrValue(gtl::ArraySlice<double> value, AttrValue* out);
77 void SetAttrValue(gtl::ArraySlice<bool> value, AttrValue* out);
81 void SetAttrValue(gtl::ArraySlice<TensorShape> value, AttrValue* out);
82 void SetAttrValue(gtl::ArraySlice<TensorShapeProto> value, AttrValue* out);
[all …]
Dnode_def_builder.h82 NodeDefBuilder& Input(gtl::ArraySlice<NodeOut> src_list);
108 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<StringPiece> value);
109 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<const char*> value);
110 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<string> value);
111 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<int32> value);
112 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<int64> value);
113 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<float> value);
114 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<bool> value);
116 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<DataType> value);
117 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<TensorShape> value);
[all …]
Dfunction.h105 gtl::ArraySlice<std::pair<string, AttrValueWrapper>> attrs);
133 const string& function_name, gtl::ArraySlice<string> in_def,
134 gtl::ArraySlice<string> out_def, gtl::ArraySlice<string> attr_def,
135 gtl::ArraySlice<Node> node_def,
136 gtl::ArraySlice<std::pair<string, string>> ret_def,
137 gtl::ArraySlice<std::pair<string, string>> control_ret_def);
144 gtl::ArraySlice<string> in_def,
145 gtl::ArraySlice<string> out_def,
146 gtl::ArraySlice<string> attr_def,
147 gtl::ArraySlice<Node> node_def,
[all …]
Dnode_def_builder.cc109 NodeDefBuilder& NodeDefBuilder::Input(gtl::ArraySlice<NodeOut> src_list) { in Input()
137 gtl::ArraySlice<NodeOut> src_list) { in ListInput()
291 ATTR(gtl::ArraySlice<StringPiece>)
292 ATTR(gtl::ArraySlice<const char*>)
293 ATTR(gtl::ArraySlice<string>)
294 ATTR(gtl::ArraySlice<int32>)
295 ATTR(gtl::ArraySlice<int64>)
296 ATTR(gtl::ArraySlice<float>)
297 ATTR(gtl::ArraySlice<bool>)
299 ATTR(gtl::ArraySlice<DataType>)
[all …]
Dnode_def_util.h85 void AddNodeAttr(StringPiece name, gtl::ArraySlice<StringPiece> value,
87 void AddNodeAttr(StringPiece name, gtl::ArraySlice<const char*> value,
89 void AddNodeAttr(StringPiece name, gtl::ArraySlice<string> value,
91 void AddNodeAttr(StringPiece name, gtl::ArraySlice<int32> value,
93 void AddNodeAttr(StringPiece name, gtl::ArraySlice<int64> value,
95 void AddNodeAttr(StringPiece name, gtl::ArraySlice<float> value,
97 void AddNodeAttr(StringPiece name, gtl::ArraySlice<bool> value,
101 void AddNodeAttr(StringPiece name, gtl::ArraySlice<DataType> value,
103 void AddNodeAttr(StringPiece name, gtl::ArraySlice<TensorShape> value,
105 void AddNodeAttr(StringPiece name, gtl::ArraySlice<PartialTensorShape> value,
[all …]
Dtensor_shape.h170 explicit TensorShapeBase(gtl::ArraySlice<int64> dim_sizes);
172 : TensorShapeBase(gtl::ArraySlice<int64>(dim_sizes)) {} in TensorShapeBase()
259 void InitDims(gtl::ArraySlice<int64> dim_sizes);
378 static Status MakeShape(gtl::ArraySlice<int32> shape, TensorShape* out);
379 static Status MakeShape(gtl::ArraySlice<int64> shape, TensorShape* out);
382 static Status MakeShape(gtl::ArraySlice<int32> shape,
384 static Status MakeShape(gtl::ArraySlice<int64> shape,
387 static string ShapeListString(const gtl::ArraySlice<TensorShape>& shapes);
398 static Status NumElements(gtl::ArraySlice<int64> shape, int64* num_elements);
453 const gtl::ArraySlice<PartialTensorShape>& shapes);
[all …]
Dfunction_testlib.h51 StringPiece name, StringPiece op, gtl::ArraySlice<string> inputs,
52 gtl::ArraySlice<std::pair<string, FunctionDefHelper::AttrValueWrapper>>
57 GraphDef GDef(gtl::ArraySlice<NodeDef> nodes,
58 gtl::ArraySlice<FunctionDef> funcs = {});
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_test_utils.cc34 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeFilterNode()
35 {"output_types", gtl::ArraySlice<TensorShape>{}}}); in MakeFilterNode()
49 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeMapAndBatchNode()
50 {"output_types", gtl::ArraySlice<TensorShape>{}}}); in MakeMapAndBatchNode()
59 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeMapNode()
60 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeMapNode()
76 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeParallelInterleaveNode()
77 {"output_types", gtl::ArraySlice<TensorShape>{}}, in MakeParallelInterleaveNode()
91 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeParallelMapNode()
92 {"output_types", gtl::ArraySlice<DataType>{}}, in MakeParallelMapNode()
[all …]
Dmap_vectorization_test.cc82 {kAttrNameTarguments, gtl::ArraySlice<DataType>({})}, in AddMapNode()
83 {kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddMapNode()
84 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{}})}, in AddMapNode()
94 {kAttrNameTarguments, gtl::ArraySlice<DataType>({})}, in AddMapNode()
95 {kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddMapNode()
96 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{}})}, in AddMapNode()
112 {{kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddPrefetchNode()
113 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{}})}}); in AddPrefetchNode()
131 {{kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddBatchNode()
132 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{10, 1}})}}); in AddBatchNode()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_impl.h42 const gtl::ArraySlice<int64>& begin,
43 const gtl::ArraySlice<int64>& end,
44 const gtl::ArraySlice<int64>& strides,
50 const gtl::ArraySlice<int64>& begin,
51 const gtl::ArraySlice<int64>& end,
52 const gtl::ArraySlice<int64>& strides,
59 void operator()(OpKernelContext* context, const gtl::ArraySlice<int64>& begin,
60 const gtl::ArraySlice<int64>& end,
61 const gtl::ArraySlice<int64>& strides,
80 const gtl::ArraySlice<int64>& begin, in HandleStridedSliceCase()
[all …]
Dtranspose_op.h32 gtl::ArraySlice<int32> perm, Tensor* out) = 0;
42 gtl::ArraySlice<int32> perm, Tensor* out) override;
52 gtl::ArraySlice<int32> perm, Tensor* out) override;
62 gtl::ArraySlice<int32> perm, Tensor* out) override;
72 gtl::ArraySlice<int32> perm, Tensor* out) override;
84 gtl::ArraySlice<int32> perm, Tensor* out) override;
96 gtl::ArraySlice<int32> perm, Tensor* out) override;
108 gtl::ArraySlice<int32> perm, Tensor* out) override;
120 gtl::ArraySlice<int32> perm, Tensor* out) override;
Drange_sampler.h78 gtl::ArraySlice<int64> extras,
89 gtl::ArraySlice<int64> extras,
91 gtl::ArraySlice<int64> avoided_values) const;
97 virtual void Update(gtl::ArraySlice<int64> values) { in Update()
130 gtl::ArraySlice<int64> extras,
132 gtl::ArraySlice<int64> avoided_values) const override;
174 void Update(gtl::ArraySlice<int64> values) override;
195 gtl::ArraySlice<int64> extras,
197 gtl::ArraySlice<int64> avoided_values) const override;
200 void Update(gtl::ArraySlice<int64> values) override;
Drange_sampler.cc32 using gtl::ArraySlice;
41 gtl::ArraySlice<int64>(), gtl::MutableArraySlice<float>()); in SampleBatch()
47 gtl::ArraySlice<int64> extras, in SampleBatchGetExpectedCount()
51 gtl::ArraySlice<int64>()); in SampleBatchGetExpectedCount()
79 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid()
81 ArraySlice<int64> avoided_values) const { in SampleBatchGetExpectedCountAvoid()
126 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid()
128 ArraySlice<int64> avoided_values) const { in SampleBatchGetExpectedCountAvoid()
190 void ThreadUnsafeUnigramSampler::Update(ArraySlice<int64> values) { in Update()
218 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid()
[all …]
Dtranspose_util_test.cc26 const gtl::ArraySlice<int32>& perm, in TestDimensionReduction()
27 const gtl::ArraySlice<int32>& expected_perm, in TestDimensionReduction()
28 const gtl::ArraySlice<int64>& expected_dims) { in TestDimensionReduction()
33 gtl::ArraySlice<int32> computed_perm(new_perm); in TestDimensionReduction()
34 gtl::ArraySlice<int64> computed_dims(new_dims); in TestDimensionReduction()
Dsparse_dense_binary_op_shared_test.cc101 const gtl::ArraySlice<int64> indices(in); in TEST_F()
103 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
130 const gtl::ArraySlice<int64> indices(in); in TEST_F()
132 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
157 const gtl::ArraySlice<int64> indices(in); in TEST_F()
159 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
189 const gtl::ArraySlice<int64> indices(in); in TEST_F()
191 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
245 test::FillValues(&shape, gtl::ArraySlice<int64>({B, M, N})); in MakeSparseTensor()
Dtranspose_functor_gpu.cu.cc58 const gtl::ArraySlice<int32> perm, Tensor* out) { in TransposeSimple()
98 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
147 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
159 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
180 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
206 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
216 const gtl::ArraySlice<int32> perm, Tensor* out) { in DoTranspose()
221 const gtl::ArraySlice<int32> perm, Tensor* out) { in DoConjugateTranspose()
Dtranspose_functor_cpu.cc36 const gtl::ArraySlice<int32> perm, Tensor* out) { in TransposeSimple()
73 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
113 const gtl::ArraySlice<int32> perm, Tensor* out) { \
119 const gtl::ArraySlice<int32> perm, \
145 const gtl::ArraySlice<int32> perm, bool conjugate, in TransposeSYCL()
183 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
191 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
Dtranspose_functor.h36 const gtl::ArraySlice<int32> perm, Tensor* out);
47 const gtl::ArraySlice<int32> perm, Tensor* out);
64 const gtl::ArraySlice<int32> perm, Tensor* out);
79 gtl::ArraySlice<int32> perm, in ReduceTransposeDimensions()
148 const gtl::ArraySlice<int32> perm, bool conjugate, in TransposeUsingEigen()
167 const gtl::ArraySlice<int32> perm, bool conjugate, in DoTransposeImpl()
252 const gtl::ArraySlice<int32> perm, bool conjugate,
/external/clang/test/SemaTemplate/
Drecovery-crash.cpp48 template <typename> class ArraySlice {}; class
51 void MemberFunction(ArraySlice<Foo>, int);
52 template <class T> void MemberFuncTemplate(ArraySlice<T>, int);
54 void NonTemplateClass::MemberFunction(ArraySlice<Foo> resource_data, in MemberFunction()
60 void NonTemplateClass::MemberFuncTemplate(ArraySlice<T> resource_data, int) { in MemberFuncTemplate()
/external/tensorflow/tensorflow/c/eager/
Dtape.h86 gtl::ArraySlice<Gradient*> gradient_tensors) const = 0;
97 gtl::ArraySlice<Gradient*> output_gradients,
124 bool ShouldRecord(gtl::ArraySlice<int64> tensor_ids,
125 gtl::ArraySlice<tensorflow::DataType> dtypes);
131 gtl::ArraySlice<int64> input_tensor_id,
132 gtl::ArraySlice<tensorflow::DataType> input_dtypes,
144 const gtl::ArraySlice<int64> target_tensor_ids,
145 const gtl::ArraySlice<int64> source_tensor_ids,
147 gtl::ArraySlice<Gradient*> output_gradients,
186 gtl::ArraySlice<int64> tensor_ids, in ShouldRecord()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dgradients.cc101 static Node* AddSymGrad(Graph* g, Node* n, gtl::ArraySlice<NodeOut> grads) { in AddSymGrad()
153 SymbolicGradientBuilder(gtl::ArraySlice<NodeOut> y_node_outputs,
154 gtl::ArraySlice<NodeOut> x_node_outputs,
155 gtl::ArraySlice<NodeOut> y_grad_node_outputs,
162 gtl::ArraySlice<NodeOut> y_node_outputs_;
163 gtl::ArraySlice<NodeOut> x_node_outputs_;
164 gtl::ArraySlice<NodeOut> y_grad_node_outputs_;
207 gtl::ArraySlice<NodeOut> y_node_outputs, in SymbolicGradientBuilder()
208 gtl::ArraySlice<NodeOut> x_node_outputs, in SymbolicGradientBuilder()
209 gtl::ArraySlice<NodeOut> y_grad_node_outputs, in SymbolicGradientBuilder()
[all …]
Dgradients.h50 Status AddSymbolicGradients(gtl::ArraySlice<NodeOut> y_node_outputs,
51 gtl::ArraySlice<NodeOut> x_node_outputs,
52 gtl::ArraySlice<NodeOut> y_grad_node_outputs,
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.h102 gtl::ArraySlice<string> serialized,
103 gtl::ArraySlice<string> example_names,
119 gtl::ArraySlice<string> serialized, gtl::ArraySlice<string> example_names,
/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops.cc47 BroadcastRange(T* data_pointer, gtl::ArraySlice<int64> data_shape, in BroadcastRange()
48 const U* cdf_pointer, gtl::ArraySlice<int64> cdf_shape) in BroadcastRange()
183 gtl::ArraySlice<int64> data_shape, in RangeEncodeImpl()
185 gtl::ArraySlice<int64> cdf_shape, string* output) const { in RangeEncodeImpl()
275 gtl::ArraySlice<int64> output_shape, in RangeDecodeImpl()
277 gtl::ArraySlice<int64> cdf_shape, in RangeDecodeImpl()
287 static_cast<gtl::ArraySlice<int32>::size_type>(cdf.dimension(1)); in RangeDecodeImpl()
298 *data = decoder.Decode(gtl::ArraySlice<int32>{cdf_slice, chip_size}); in RangeDecodeImpl()
/external/tensorflow/tensorflow/core/lib/gtl/
Dedit_distance.h47 inline int64 LevenshteinDistance(const gtl::ArraySlice<T>& s, in LevenshteinDistance()
48 const gtl::ArraySlice<T>& t, const Cmp& cmp) { in LevenshteinDistance()
100 gtl::ArraySlice<typename Container1::value_type>(s.data(), s.size()), in LevenshteinDistance()
101 gtl::ArraySlice<typename Container1::value_type>(t.data(), t.size()), in LevenshteinDistance()

12345678