Home
last modified time | relevance | path

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

12345678910

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_test_utils.cc37 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeBatchV2Node()
38 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeBatchV2Node()
50 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeParallelBatchNode()
51 {"output_types", gtl::ArraySlice<DataType>{}}, in MakeParallelBatchNode()
66 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeCacheV2Node()
67 {"output_types", gtl::ArraySlice<DataType>{}}, in MakeCacheV2Node()
77 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeFilterNode()
78 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeFilterNode()
92 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in MakeMapAndBatchNode()
93 {"output_types", gtl::ArraySlice<DataType>{}}}); in MakeMapAndBatchNode()
[all …]
Dinject_prefetch_test.cc67 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST_P()
68 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST_P()
93 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST()
94 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST()
111 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST()
112 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST()
131 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST()
132 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST()
135 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST()
136 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST()
[all …]
Dauto_shard_test.cc51 {"output_shapes", gtl::ArraySlice<TensorShape>{}}, in FinishItem()
52 {"output_types", gtl::ArraySlice<DataType>{}}}); in FinishItem()
57 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in FinishItem()
58 {"output_types", gtl::ArraySlice<DataType>{}}}); in FinishItem()
63 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in FinishItem()
64 {"output_types", gtl::ArraySlice<DataType>{}}}); in FinishItem()
83 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST()
84 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST()
113 {{"output_shapes", gtl::ArraySlice<TensorShape>{}}, in TEST()
114 {"output_types", gtl::ArraySlice<DataType>{}}}), in TEST()
[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_t> 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_t> 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_t>)
315 ATTR(gtl::ArraySlice<float>)
316 ATTR(gtl::ArraySlice<bool>)
[all …]
Dfunction.h119 gtl::ArraySlice<std::pair<string, AttrValueWrapper>> attrs);
164 const std::string& function_name, gtl::ArraySlice<string> in_def,
165 gtl::ArraySlice<string> out_def, gtl::ArraySlice<string> attr_def,
166 gtl::ArraySlice<Node> node_def,
167 gtl::ArraySlice<std::pair<string, string>> ret_def,
168 gtl::ArraySlice<std::pair<string, string>> control_ret_def);
175 gtl::ArraySlice<string> in_def,
176 gtl::ArraySlice<string> out_def,
177 gtl::ArraySlice<string> attr_def,
178 gtl::ArraySlice<Node> node_def,
[all …]
Dtensor_shape.h168 explicit TensorShapeBase(gtl::ArraySlice<int64_t> dim_sizes);
170 : TensorShapeBase(gtl::ArraySlice<int64_t>(dim_sizes)) {} in TensorShapeBase()
183 static Status BuildTensorShapeBase(gtl::ArraySlice<int64_t> dim_sizes,
187 return BuildTensorShapeBase(gtl::ArraySlice<int64_t>(dim_sizes), out); in BuildTensorShapeBase()
327 Status InitDims(gtl::ArraySlice<int64_t> dim_sizes);
366 static Status BuildTensorShape(gtl::ArraySlice<int64_t> dim_sizes, in BuildTensorShape()
372 return BuildTensorShape(gtl::ArraySlice<int64_t>(dim_sizes), out); in BuildTensorShape()
504 static Status MakeShape(gtl::ArraySlice<int32> shape, TensorShape* out);
505 static Status MakeShape(gtl::ArraySlice<int64_t> shape, TensorShape* out);
510 static Status MakeShape(gtl::ArraySlice<int32> shape,
[all …]
Dkernel_def_builder.cc39 const char* attr_name, gtl::ArraySlice<int64_t> allowed) { in AttrConstraint()
54 gtl::ArraySlice<int64_t>(std::initializer_list<int64_t>({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.h104 void AddNodeAttr(StringPiece name, gtl::ArraySlice<StringPiece> value,
106 void AddNodeAttr(StringPiece name, gtl::ArraySlice<const char*> value,
108 void AddNodeAttr(StringPiece name, gtl::ArraySlice<string> value,
110 void AddNodeAttr(StringPiece name, gtl::ArraySlice<int32> value,
112 void AddNodeAttr(StringPiece name, gtl::ArraySlice<int64_t> value,
114 void AddNodeAttr(StringPiece name, gtl::ArraySlice<float> value,
116 void AddNodeAttr(StringPiece name, gtl::ArraySlice<bool> value,
120 void AddNodeAttr(StringPiece name, gtl::ArraySlice<DataType> value,
122 void AddNodeAttr(StringPiece name, gtl::ArraySlice<TensorShape> value,
124 void AddNodeAttr(StringPiece name, gtl::ArraySlice<PartialTensorShape> value,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_impl.h41 const gtl::ArraySlice<int64_t>& begin,
42 const gtl::ArraySlice<int64_t>& end,
43 const gtl::ArraySlice<int64_t>& strides,
49 const gtl::ArraySlice<int64_t>& begin,
50 const gtl::ArraySlice<int64_t>& end,
51 const gtl::ArraySlice<int64_t>& strides,
59 const gtl::ArraySlice<int64_t>& begin,
60 const gtl::ArraySlice<int64_t>& end,
61 const gtl::ArraySlice<int64_t>& strides,
79 const gtl::ArraySlice<int64_t>& begin, in HandleStridedSliceCase()
[all …]
Drange_sampler.h78 gtl::ArraySlice<int64_t> extras,
89 gtl::ArraySlice<int64_t> extras,
91 gtl::ArraySlice<int64_t> avoided_values) const;
97 virtual void Update(gtl::ArraySlice<int64_t> values) { in Update()
130 gtl::ArraySlice<int64_t> extras,
132 gtl::ArraySlice<int64_t> avoided_values) const override;
174 void Update(gtl::ArraySlice<int64_t> values) override;
195 gtl::ArraySlice<int64_t> extras,
197 gtl::ArraySlice<int64_t> avoided_values) const override;
200 void Update(gtl::ArraySlice<int64_t> 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_t>(), gtl::MutableArraySlice<float>()); in SampleBatch()
49 gtl::ArraySlice<int64_t> extras, in SampleBatchGetExpectedCount()
53 gtl::ArraySlice<int64_t>()); in SampleBatchGetExpectedCount()
81 MutableArraySlice<float> batch_expected_count, ArraySlice<int64_t> extras, in SampleBatchGetExpectedCountAvoid()
83 ArraySlice<int64_t> avoided_values) const { in SampleBatchGetExpectedCountAvoid()
128 MutableArraySlice<float> batch_expected_count, ArraySlice<int64_t> extras, in SampleBatchGetExpectedCountAvoid()
130 ArraySlice<int64_t> avoided_values) const { in SampleBatchGetExpectedCountAvoid()
194 void ThreadUnsafeUnigramSampler::Update(ArraySlice<int64_t> values) { in Update()
222 MutableArraySlice<float> batch_expected_count, ArraySlice<int64_t> 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_t> expected_dims) { in TestDimensionReduction()
33 gtl::ArraySlice<int32> computed_perm(new_perm); in TestDimensionReduction()
34 gtl::ArraySlice<int64_t> computed_dims(new_dims); in TestDimensionReduction()
Dsparse_dense_binary_op_shared_test.cc102 const gtl::ArraySlice<int64_t> indices(in); in TEST_F()
104 const gtl::ArraySlice<int64_t> shape(sh); in TEST_F()
131 const gtl::ArraySlice<int64_t> indices(in); in TEST_F()
133 const gtl::ArraySlice<int64_t> shape(sh); in TEST_F()
158 const gtl::ArraySlice<int64_t> indices(in); in TEST_F()
160 const gtl::ArraySlice<int64_t> shape(sh); in TEST_F()
190 const gtl::ArraySlice<int64_t> indices(in); in TEST_F()
192 const gtl::ArraySlice<int64_t> shape(sh); in TEST_F()
246 test::FillValues(&shape, gtl::ArraySlice<int64_t>({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_t> dim_range, const int64_t isd);
/external/tensorflow/tensorflow/core/data/service/
Dtest_util.cc64 {"output_shapes", gtl::ArraySlice<TensorShape>{TensorShape()}}, in GetMapNode()
65 {"output_types", gtl::ArraySlice<DataType>{DT_INT64}}}); in GetMapNode()
101 {{"output_shapes", gtl::ArraySlice<TensorShape>{TensorShape()}}, in RangeDataset()
102 {"output_types", gtl::ArraySlice<DataType>{DT_INT64}}}), in RangeDataset()
119 {{"output_shapes", gtl::ArraySlice<TensorShape>{TensorShape()}}, in RangeSquareDataset()
120 {"output_types", gtl::ArraySlice<DataType>{DT_INT64}}}), in RangeSquareDataset()
138 {{"output_shapes", gtl::ArraySlice<TensorShape>{TensorShape()}}, in RangeDatasetWithShardHint()
139 {"output_types", gtl::ArraySlice<DataType>{DT_INT64}}}), in RangeDatasetWithShardHint()
146 {{"output_shapes", gtl::ArraySlice<TensorShape>{TensorShape()}}, in RangeDatasetWithShardHint()
147 {"output_types", gtl::ArraySlice<DataType>{DT_INT64}}}), in RangeDatasetWithShardHint()
[all …]
/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/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_t> tensor_ids,
142 gtl::ArraySlice<tensorflow::DataType> dtypes) const;
159 gtl::ArraySlice<int64_t> input_tensor_id,
160 gtl::ArraySlice<tensorflow::DataType> input_dtypes,
175 const gtl::ArraySlice<int64_t> target_tensor_ids,
176 const gtl::ArraySlice<int64_t> source_tensor_ids,
178 gtl::ArraySlice<Gradient*> output_gradients, absl::Span<Gradient*> result,
284 gtl::ArraySlice<int64_t> input_tensor_id,
[all …]
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dconcat_split_util.h38 Status Concat(OpKernelContext* context, const gtl::ArraySlice<Tensor> inputs, in Concat()
95 const gtl::ArraySlice<Tensor> inputs, Tensor* output) { in Concat()
121 const gtl::ArraySlice<int64_t> sizes, in SplitEasyCases()
158 const gtl::ArraySlice<int64_t> sizes, in SplitCPU()
201 const gtl::ArraySlice<int64_t>& sizes, in SplitGPU()
212 const gtl::ArraySlice<int64_t> sizes, in Split()
231 const gtl::ArraySlice<int64_t> sizes, in Split()
/external/tensorflow/tensorflow/core/lib/gtl/
Dedit_distance.h47 inline int64_t LevenshteinDistance(const gtl::ArraySlice<T>& s, in LevenshteinDistance()
48 const gtl::ArraySlice<T>& t, in LevenshteinDistance()
101 gtl::ArraySlice<typename Container1::value_type>(s.data(), s.size()), in LevenshteinDistance()
102 gtl::ArraySlice<typename Container1::value_type>(t.data(), t.size()), in LevenshteinDistance()

12345678910