Home
last modified time | relevance | path

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

12345678910

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_test_utils.cc36 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeBatchV2Node()
37 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeBatchV2Node()
51 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeCacheV2Node()
52 {"output_types", gtl::ArraySlice<DataType>{}}, in MakeCacheV2Node()
62 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeFilterNode()
63 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeFilterNode()
77 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeMapAndBatchNode()
78 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeMapAndBatchNode()
87 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeMapNode()
88 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeMapNode()
[all …]
Dmap_vectorization_test.cc95 {kAttrNameTarguments, gtl::ArraySlice<DataType>({})}, in AddMapNode()
96 {kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddMapNode()
97 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{}})}, in AddMapNode()
107 {kAttrNameTarguments, gtl::ArraySlice<DataType>({})}, in AddMapNode()
108 {kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddMapNode()
109 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{}})}, in AddMapNode()
125 {{kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddPrefetchNode()
126 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{}})}}); in AddPrefetchNode()
144 {{kAttrNameOutputTypes, gtl::ArraySlice<DataType>({DT_INT64})}, in AddBatchNode()
145 {kAttrNameOutputShapes, gtl::ArraySlice<TensorShape>({{10, 1}})}}); in AddBatchNode()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.h71 void SetAttrValue(gtl::ArraySlice<string> value, AttrValue* out);
72 void SetAttrValue(gtl::ArraySlice<tstring> value, AttrValue* out);
73 void SetAttrValue(gtl::ArraySlice<const char*> value, AttrValue* out);
74 void SetAttrValue(gtl::ArraySlice<StringPiece> value, AttrValue* out);
75 void SetAttrValue(gtl::ArraySlice<int64> value, AttrValue* out);
76 void SetAttrValue(gtl::ArraySlice<int32> value, AttrValue* out);
77 void SetAttrValue(gtl::ArraySlice<float> value, AttrValue* out);
78 void SetAttrValue(gtl::ArraySlice<double> value, AttrValue* out);
79 void SetAttrValue(gtl::ArraySlice<bool> value, AttrValue* out);
83 void SetAttrValue(gtl::ArraySlice<TensorShape> value, AttrValue* out);
[all …]
Dnode_def_builder.h84 NodeDefBuilder& Input(gtl::ArraySlice<NodeOut> src_list);
111 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<StringPiece> value);
112 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<const char*> value);
113 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<string> value);
114 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<tstring> value);
115 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<int32> value);
116 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<int64> value);
117 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<float> value);
118 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<bool> value);
120 NodeDefBuilder& Attr(StringPiece name, gtl::ArraySlice<DataType> value);
[all …]
Dnode_def_builder.cc109 NodeDefBuilder& NodeDefBuilder::Input(gtl::ArraySlice<NodeOut> src_list) { in Input()
137 gtl::ArraySlice<NodeOut> src_list) { in ListInput()
309 ATTR(gtl::ArraySlice<StringPiece>)
310 ATTR(gtl::ArraySlice<const char*>)
311 ATTR(gtl::ArraySlice<string>)
312 ATTR(gtl::ArraySlice<tstring>)
313 ATTR(gtl::ArraySlice<int32>)
314 ATTR(gtl::ArraySlice<int64>)
315 ATTR(gtl::ArraySlice<float>)
316 ATTR(gtl::ArraySlice<bool>)
[all …]
Dfunction.h118 gtl::ArraySlice<std::pair<string, AttrValueWrapper>> attrs);
146 const std::string& function_name, gtl::ArraySlice<string> in_def,
147 gtl::ArraySlice<string> out_def, gtl::ArraySlice<string> attr_def,
148 gtl::ArraySlice<Node> node_def,
149 gtl::ArraySlice<std::pair<string, string>> ret_def,
150 gtl::ArraySlice<std::pair<string, string>> control_ret_def);
157 gtl::ArraySlice<string> in_def,
158 gtl::ArraySlice<string> out_def,
159 gtl::ArraySlice<string> attr_def,
160 gtl::ArraySlice<Node> node_def,
[all …]
Dkernel_def_builder.cc38 const char* attr_name, gtl::ArraySlice<int64> allowed) { in AttrConstraint()
54 gtl::ArraySlice<int64>(std::initializer_list<int64>({allowed}))); in AttrConstraint()
59 const char* attr_name, gtl::ArraySlice<string> allowed) { in AttrConstraint()
74 gtl::ArraySlice<string>(std::initializer_list<string>({allowed}))); in AttrConstraint()
79 const char* attr_name, gtl::ArraySlice<const char*> allowed) { in AttrConstraint()
93 gtl::ArraySlice<const char*>( in AttrConstraint()
108 const char* attr_name, gtl::ArraySlice<DataType> allowed) { in TypeConstraint()
Dnode_def_util.h99 void AddNodeAttr(StringPiece name, gtl::ArraySlice<StringPiece> value,
101 void AddNodeAttr(StringPiece name, gtl::ArraySlice<const char*> value,
103 void AddNodeAttr(StringPiece name, gtl::ArraySlice<string> value,
105 void AddNodeAttr(StringPiece name, gtl::ArraySlice<int32> value,
107 void AddNodeAttr(StringPiece name, gtl::ArraySlice<int64> value,
109 void AddNodeAttr(StringPiece name, gtl::ArraySlice<float> value,
111 void AddNodeAttr(StringPiece name, gtl::ArraySlice<bool> value,
115 void AddNodeAttr(StringPiece name, gtl::ArraySlice<DataType> value,
117 void AddNodeAttr(StringPiece name, gtl::ArraySlice<TensorShape> value,
119 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()
184 static Status BuildTensorShapeBase(gtl::ArraySlice<int64> dim_sizes,
188 return BuildTensorShapeBase(gtl::ArraySlice<int64>(dim_sizes), out); in BuildTensorShapeBase()
327 Status InitDims(gtl::ArraySlice<int64> dim_sizes);
481 static Status MakeShape(gtl::ArraySlice<int32> shape, TensorShape* out);
482 static Status MakeShape(gtl::ArraySlice<int64> shape, TensorShape* out);
485 static Status MakeShape(gtl::ArraySlice<int32> shape,
487 static Status MakeShape(gtl::ArraySlice<int64> shape,
491 const gtl::ArraySlice<TensorShape>& shapes);
[all …]
Dfunction_testlib.h59 StringPiece name, StringPiece op, gtl::ArraySlice<string> inputs,
60 gtl::ArraySlice<std::pair<string, FunctionDefHelper::AttrValueWrapper>>
65 GraphDef GDef(gtl::ArraySlice<NodeDef> nodes,
66 gtl::ArraySlice<FunctionDef> funcs = {});
/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 …]
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;
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;
74 gtl::ArraySlice<int32> perm, Tensor* out) override;
86 gtl::ArraySlice<int32> perm, Tensor* out) override;
98 gtl::ArraySlice<int32> perm, Tensor* out) override;
Drange_sampler.cc33 using gtl::ArraySlice;
42 gtl::ArraySlice<int64>(), gtl::MutableArraySlice<float>()); in SampleBatch()
48 gtl::ArraySlice<int64> extras, in SampleBatchGetExpectedCount()
52 gtl::ArraySlice<int64>()); in SampleBatchGetExpectedCount()
80 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid()
82 ArraySlice<int64> avoided_values) const { in SampleBatchGetExpectedCountAvoid()
127 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid()
129 ArraySlice<int64> avoided_values) const { in SampleBatchGetExpectedCountAvoid()
193 void ThreadUnsafeUnigramSampler::Update(ArraySlice<int64> values) { in Update()
221 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.cc102 const gtl::ArraySlice<int64> indices(in); in TEST_F()
104 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
131 const gtl::ArraySlice<int64> indices(in); in TEST_F()
133 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
158 const gtl::ArraySlice<int64> indices(in); in TEST_F()
160 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
190 const gtl::ArraySlice<int64> indices(in); in TEST_F()
192 const gtl::ArraySlice<int64> shape(sh); in TEST_F()
246 test::FillValues(&shape, gtl::ArraySlice<int64>({B, M, N})); in MakeSparseTensor()
Dtranspose_functor_gpu.cu.cc59 const gtl::ArraySlice<int32> perm, Tensor* out) { in TransposeSimple()
99 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
148 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
160 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
181 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
207 const gtl::ArraySlice<int32> perm, Tensor* out) { in run()
217 const gtl::ArraySlice<int32> perm, Tensor* out) { in DoTranspose()
222 const gtl::ArraySlice<int32> perm, Tensor* out) { in DoConjugateTranspose()
Droll_op.h37 const int num_dims, const gtl::ArraySlice<int32> dim_size,
39 const gtl::ArraySlice<int32> threshold,
40 const gtl::ArraySlice<int64> dim_range, const int64 isd);
/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/llvm-project/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/core/common_runtime/
Dgradients.cc102 static Node* AddSymGrad(Graph* g, Node* n, gtl::ArraySlice<NodeOut> grads) { in AddSymGrad()
154 SymbolicGradientBuilder(gtl::ArraySlice<NodeOut> y_node_outputs,
155 gtl::ArraySlice<NodeOut> x_node_outputs,
156 gtl::ArraySlice<NodeOut> y_grad_node_outputs,
163 gtl::ArraySlice<NodeOut> y_node_outputs_;
164 gtl::ArraySlice<NodeOut> x_node_outputs_;
165 gtl::ArraySlice<NodeOut> y_grad_node_outputs_;
211 gtl::ArraySlice<NodeOut> y_node_outputs, in SymbolicGradientBuilder()
212 gtl::ArraySlice<NodeOut> x_node_outputs, in SymbolicGradientBuilder()
213 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/c/eager/
Dtape.h93 gtl::ArraySlice<Gradient*> gradient_tensors) const = 0;
102 gtl::ArraySlice<Gradient*> output_gradients,
141 bool ShouldRecord(gtl::ArraySlice<int64> tensor_ids,
142 gtl::ArraySlice<tensorflow::DataType> dtypes) const;
159 gtl::ArraySlice<int64> input_tensor_id,
160 gtl::ArraySlice<tensorflow::DataType> input_dtypes,
175 const gtl::ArraySlice<int64> target_tensor_ids,
176 const gtl::ArraySlice<int64> source_tensor_ids,
178 gtl::ArraySlice<Gradient*> output_gradients, absl::Span<Gradient*> result,
284 gtl::ArraySlice<int64> input_tensor_id,
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dmodule_spec.h33 port::ArraySlice<const uint8> cuda_cubin_in_memory() const { in cuda_cubin_in_memory()
44 void AddCudaCubinInMemory(port::ArraySlice<const uint8> cubin_bytes) { in AddCudaCubinInMemory()
57 port::ArraySlice<const uint8> cuda_cubin_in_memory_;
/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()

12345678910