Home
last modified time | relevance | path

Searched refs:output_dim (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/layers/
Dembeddings.py108 output_dim, argument
121 if input_dim <= 0 or output_dim <= 0:
124 input_dim, output_dim))
138 self.output_dim = output_dim
150 shape=(self.input_dim, self.output_dim),
166 return input_shape + (self.output_dim,)
185 return (input_shape[0],) + tuple(in_lens) + (self.output_dim,)
201 'output_dim': self.output_dim,
Drecurrent.py170 output_dim = cell.output_size
172 output_dim = cell.state_size[0]
174 output_dim = cell.state_size
176 tensor_shape.TensorShape(output_dim).as_list())
490 output_dim = tensor_shape.TensorShape(flat_output_size).as_list()
494 [time_step, batch] + output_dim)
497 [batch, time_step] + output_dim)
499 output_shape = tensor_shape.TensorShape([batch] + output_dim)
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconv_grad_size_util.cc70 TF_ASSIGN_OR_RETURN(SpatialDimensionOutputSizeAndPadding output_dim, in ConvGradExtractAndVerifyDimension()
73 if (output_size != output_dim.output_size) { in ConvGradExtractAndVerifyDimension()
76 output_size, ", computed = ", output_dim.output_size, in ConvGradExtractAndVerifyDimension()
83 dim.output_size = (output_dim.output_size - 1) * stride + 1; in ConvGradExtractAndVerifyDimension()
85 dim.pad_before = effective_filter_size - 1 - output_dim.pad_before; in ConvGradExtractAndVerifyDimension()
Dmatrix.cc500 if (auto output_dim = output_dimension_number(x_config[d])) { in Einsum() local
501 output_transpose_dims.push_back(*output_dim); in Einsum()
508 if (auto output_dim = output_dimension_number(y_config[d])) { in Einsum() local
509 output_transpose_dims.push_back(*output_dim); in Einsum()
/external/tensorflow/tensorflow/core/util/
Dbcast.h202 int output_dim = -1; in BCastList() local
206 output_dim = -1; in BCastList()
217 if (!output_dim_set || copy[i][j] == output_dim) { in BCastList()
218 output_dim = copy[i][j]; in BCastList()
226 output_.push_back(output_dim_set ? output_dim : 1); in BCastList()
262 result_.back() = mul_dims(result_.back(), output_dim); in BCastList()
266 mul_dims(bcast_[i].back(), current_is_one[i] ? output_dim : 1); in BCastList()
272 result_.push_back(output_dim); in BCastList()
275 bcast_[i].push_back(current_is_one[i] ? output_dim : 1); in BCastList()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc81 const int64_t output_dim = in SpaceToBatch() local
83 if (output_dim < 0) { in SpaceToBatch()
85 ctx, output_dim >= 0, in SpaceToBatch()
151 output_shape[0] = output_dim; in SpaceToBatch()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dbatch_to_space_nd_test.cc25 int input_dim, int output_dim) { in GetIndexRange() argument
29 output_dim, &index_start, &index_end); in GetIndexRange()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dpad_tester.cc42 int32_t output_dim = InputShape()[i]; in OutputShape() local
44 output_dim += InputPrePaddings()[i]; in OutputShape()
47 output_dim += InputPostPaddings()[i]; in OutputShape()
49 output_shape.push_back(output_dim); in OutputShape()
Dquantized_pad_tester.cc41 int32_t output_dim = InputShape()[i]; in OutputShape() local
43 output_dim += InputPrePaddings()[i]; in OutputShape()
46 output_dim += InputPostPaddings()[i]; in OutputShape()
48 output_shape.push_back(output_dim); in OutputShape()
Dreshape_tester.h46 for (int32_t output_dim : output_shape) { in OutputShape() local
47 EXPECT_GT(output_dim, 0); in OutputShape()
/external/tensorflow/tensorflow/core/ops/
Dimage_ops.cc227 DimensionHandle output_dim; in CombinedNMSShapeFn() local
230 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(3, &output_dim)); in CombinedNMSShapeFn()
231 if (c->ValueKnown(output_dim) && c->Value(output_dim) <= 0) { in CombinedNMSShapeFn()
241 output_size = c->Value(output_dim); in CombinedNMSShapeFn()
248 output_size = std::min(c->Value(output_dim), in CombinedNMSShapeFn()
1026 DimensionHandle output_dim; in __anonf6f833bb1b02() local
1027 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &output_dim)); in __anonf6f833bb1b02()
1028 c->set_output(0, c->MakeShape({output_dim})); in __anonf6f833bb1b02()
1053 DimensionHandle output_dim; in __anonf6f833bb1c02() local
1054 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(2, &output_dim)); in __anonf6f833bb1c02()
[all …]
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dargmax_spmd_expander.cc158 int output_dim = 0; in ComputeLayoutBackward() local
163 layout_sharding.emplace_back(output_layout.sharding_spec(output_dim)); in ComputeLayoutBackward()
164 output_dim += 1; in ComputeLayoutBackward()
Dreduce_spmd_expander.cc301 int output_dim = 0; in ComputeLayoutBackward() local
307 if (keep_dims) output_dim += 1; in ComputeLayoutBackward()
310 output_layout.sharding_spec(output_dim)); in ComputeLayoutBackward()
311 output_dim += 1; in ComputeLayoutBackward()
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc220 for (int64_t output_dim : output_dims) { in FoldGatherOfGather() local
221 simulated_index.insert(simulated_index.begin() + output_dim, in FoldGatherOfGather()
543 for (int64_t output_dim : operand->output_dims()) { in ReshapeToAddDegenerateDims() local
544 output_dims_bitvector[output_dim] = true; in ReshapeToAddDegenerateDims()
570 for (int64_t output_dim : new_output_dims) { in ReshapeToAddDegenerateDims() local
571 EraseAt(&new_source_shape_dims, output_dim); in ReshapeToAddDegenerateDims()
717 int64_t output_dim = scalar_indexed->output_dims()[i]; in FoldReshapeOfGatherNoDegenerateDims() local
719 reshape_passthrough_dims, output_dim); in FoldReshapeOfGatherNoDegenerateDims()
881 auto is_broadcasted_dim = [&](int64_t output_dim) { in ComputeArrayForElementwiseBinaryOp() argument
882 return absl::c_find(broadcast_dims, output_dim) == broadcast_dims.end(); in ComputeArrayForElementwiseBinaryOp()
Ddynamic_padder.cc367 const int64_t output_dim = output_dims[i]; in GenerateBinaryMask() local
368 HloInstruction* dynamic_size = output_dynamic_dims[output_dim]; in GenerateBinaryMask()
371 output_shape.dimensions(output_dim)))); in GenerateBinaryMask()
573 for (int64_t output_dim : output_dims) { in RewriteDynamicReshapeSplitInput() local
575 dynamic_dimension_inference->GetDynamicSize(reshape, {}, output_dim); in RewriteDynamicReshapeSplitInput()
580 output_dim)); in RewriteDynamicReshapeSplitInput()
653 int64_t output_dim, absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeCombineInput() argument
664 ShapeUtil::MakeShape(xla::S32, {output_shape.dimensions(output_dim)}); in RewriteDynamicReshapeCombineInput()
669 GenerateBinaryMask(reshape, output_dim, input_dims, input_dynamic_dims, in RewriteDynamicReshapeCombineInput()
720 if (i != output_dim) { in RewriteDynamicReshapeCombineInput()
[all …]
Ddynamic_dimension_inference.cc1404 for (int64_t output_dim = 0; output_dim < output_rank; ++output_dim) { in HandleGather() local
1405 if (!absl::c_linear_search(gather_dims.offset_dims(), output_dim)) { in HandleGather()
1411 parent_->SetDynamicSize(hlo, {}, output_dim, dynamic_size); in HandleGather()
/external/tensorflow/tensorflow/python/debug/examples/v2/
Ddebug_mnist_v2.py160 def get_dense_weights(input_dim, output_dim): argument
164 kernel = tf.Variable(initial_kernel([input_dim, output_dim]))
165 bias = tf.Variable(tf.constant(0.1, shape=[output_dim]))
/external/tensorflow/tensorflow/python/debug/examples/v1/
Ddebug_mnist_v1.py158 def nn_layer(input_tensor, input_dim, output_dim, layer_name, act=tf.nn.relu): argument
164 weights = weight_variable([input_dim, output_dim])
166 biases = bias_variable([output_dim])
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc140 int output_dim = -1; in Invoke() local
151 if (!output_dim_set || copy[i][j] == output_dim) { in Invoke()
152 output_dim = copy[i][j]; in Invoke()
/external/XNNPACK/test/
Dconstant-pad-operator-tester.h83 inline size_t output_dim(size_t i) const { in output_dim() function
90 elements *= output_dim(i); in num_output_elements()
126 output_dims[XNN_MAX_TENSOR_DIMS - num_dims() + i] = output_dim(i); in TestX8()
241 output_dims[XNN_MAX_TENSOR_DIMS - num_dims() + i] = output_dim(i); in TestX16()
356 output_dims[XNN_MAX_TENSOR_DIMS - num_dims() + i] = output_dim(i); in TestX32()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dreplace_cast_hacks_with_tf_xla_ops.cc77 int output_dim, const SmallVector<int64_t> &weight_non_output_indices) { in CalculateZeroPointOffset() argument
86 RankedTensorType::get({output_dim}, builder.getIntegerType(32)); in CalculateZeroPointOffset()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_symbolic_shape_optimization.cc239 int64_t output_dim = out_type.getShape()[d]; in matchAndRewrite() local
242 if (output_dim >= 0) continue; in matchAndRewrite()
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dbroadcast_to_ops_test.py54 for output_dim in range(input_dim, 6):
57 output_shape = [2] * output_dim
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc412 int64_t output_dim = output_shape[spatial_dim_index]; in verify() local
413 if (!static_dims(input_dim, output_dim)) return success(); in verify()
418 if (crops_values.empty() && output_dim > input_dim * block_size) in verify()
424 << output_dim << ", input " << dim_name << " " << input_dim in verify()
431 if (output_dim != input_dim_pad - crop_a - crop_b) in verify()
437 << dim_name << " " << output_dim << ", input " << dim_name in verify()
Dtf_ops_n_z.cc2470 const int64_t output_dim = output_type.getDimSize(i); in verify() local
2480 !ShapedType::isDynamic(output_dim) && output_dim != input_dim * m) { in verify()
2484 << "output.shape[" << i << "] (" << output_dim << ")"; in verify()

12