| /external/tensorflow/tensorflow/core/grappler/costs/ |
| utils_test.cc | 83 NodeDef* input_sizes = graph.add_node(); local
84 name_to_node["input_sizes"] = input;
85 CreateConstSizesOp("input_sizes",
87 input_sizes);
116 .Input("input_sizes", 0, DT_INT32)
|
| /external/eigen/bench/tensors/ |
| tensor_benchmarks.h | 418 Eigen::array<TensorIndex, 2> input_sizes; local
419 input_sizes[0] = m_;
420 input_sizes[1] = n_;
421 TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, input_sizes);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_ops_3d.cc | 172 const Tensor& input_sizes = context->input(0); variable
174 input_sizes.vec<int32>(), &input_shape));
459 const Tensor& input_sizes = context->input(0); variable
461 input_sizes.vec<int32>(), &input_shape));
[all...] |
| depthwise_conv_grad_op.cc | 554 const Tensor& input_sizes = context->input(0); variable
557 context, TensorShapeUtils::IsVector(input_sizes.shape()),
559 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
560 input_sizes.dims()));
562 const int32* in_sizes_data = input_sizes.template flat<int32>().data();
563 for (int i = 0; i < input_sizes.NumElements(); ++i) {
566 " of input_sizes must be >= 0"));
608 .HostMemory("input_sizes"),
615 .HostMemory("input_sizes"),
[all...] |
| conv_grad_input_ops.cc | 221 const Tensor& input_sizes = context->input(0); variable
225 context, TensorShapeUtils::IsVector(input_sizes.shape()),
227 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
228 input_sizes.dims()));
231 input_sizes.vec<int32>(), &input_shape));
336 const Tensor& input_sizes = context->input(0); variable
340 context, TensorShapeUtils::IsVector(input_sizes.shape()),
342 "Conv2DBackpropInput: input_sizes input must be 1-dim, not ",
343 input_sizes.dims()));
346 input_sizes.vec<int32>(), &input_shape))
654 const Tensor& input_sizes = context->input(0); variable
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| layout_optimizer_test.cc | 91 Output input_sizes = local
112 ops::Identity(s->WithOpName("InputSizesIdentity"), input_sizes);
115 s->WithOpName("Conv2DBackpropInput"), input_sizes, filter, output,
181 auto input_sizes = GetAttrValue(*input_sizes_node); local
184 test::ExpectTensorEqual<int>(input_sizes_expected, input_sizes);
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_control_wrapper.cc | 99 std::vector<int> input_sizes; local
112 input_sizes.emplace_back(static_cast<int>(buf_size));
129 input_sizes.size(), input_sizes.data(), output_sizes.size(),
|
| /external/mesa3d/src/compiler/nir/ |
| nir.h | 743 unsigned input_sizes[4]; member in struct:__anon29275
784 if (nir_op_infos[instr->op].input_sizes[src] > 0)
785 return channel < nir_op_infos[instr->op].input_sizes[src];
799 if (nir_op_infos[instr->op].input_sizes[src] > 0)
800 return nir_op_infos[instr->op].input_sizes[src];
[all...] |
