Home
last modified time | relevance | path

Searched refs:MakeDim (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Daudio_ops.cc43 channels_dim = c->MakeDim(desired_channels); in DecodeWavShapeFn()
55 samples_dim = c->MakeDim(desired_samples); in DecodeWavShapeFn()
93 output_length = c->MakeDim(output_length_value); in SpectrogramShapeFn()
97 c->MakeDim(1 + NextPowerOfTwo(window_size) / 2); in SpectrogramShapeFn()
116 DimensionHandle output_channels = c->MakeDim(dct_coefficient_count); in MfccShapeFn()
Dimage_ops.cc55 height = c->MakeDim(vec(0)); in SetOutputToSizedImage()
56 width = c->MakeDim(vec(1)); in SetOutputToSizedImage()
82 channels_dim = c->MakeDim(channels); in DecodeImageShapeFn()
341 c->ReplaceDim(input, 1, c->MakeDim(size_vec(0)), &input)); in __anon6a71d27f0602()
343 c->ReplaceDim(input, 2, c->MakeDim(size_vec(1)), &input)); in __anon6a71d27f0602()
375 h = c->MakeDim(size_vec(0)); in __anon6a71d27f0702()
376 w = c->MakeDim(size_vec(1)); in __anon6a71d27f0702()
420 channels_dim = c->MakeDim(channels); in __anon6a71d27f0802()
430 h = c->MakeDim(crop_window_vec(2)); in __anon6a71d27f0802()
431 w = c->MakeDim(crop_window_vec(3)); in __anon6a71d27f0802()
Dset_ops.cc74 output_rank = c->MakeDim(input0_rank); in __anon967df2c80102()
79 output_rank = c->MakeDim(c->Rank(input1_shape)); in __anon967df2c80102()
122 output_rank_dim = c->MakeDim(input0_rank); in __anon967df2c80202()
Dragged_conversion_ops.cc63 ? c->MakeDim(c->Rank(rt_dense_values) + num_splits) in RaggedTensorToSparseShapeFn()
Dsdca_ops.cc38 {c->Matrix(InferenceContext::kUnknownDim, c->MakeDim(4))}); in ApplySdcaOptimizerShapeFn()
Dspectral_ops.cc103 TF_RETURN_IF_ERROR(c->ReplaceDim(out, -rank + i, c->MakeDim(dim), &out)); in RFFTShape()
Darray_ops.cc199 known_out_elems = c->MakeDim(1); in SetOutputShapeForReshape()
217 known_in_elems = c->MakeDim(1); in SetOutputShapeForReshape()
296 passed_shape, 0, c->MakeDim(shape_inference::DimensionOrConstant(1)), in __anon7c94107b0202()
346 dims.push_back(c->MakeDim(c->num_inputs())); in __anon7c94107b0302()
678 c->ReplaceDim(input, split_dim, c->MakeDim(size), &output_shape)); in __anon7c94107b0b02()
707 dims.push_back(c->MakeDim(shape.dim_size(i))); in __anon7c94107b0c02()
1000 dims[i] = c->MakeDim(std::max(h_values(i), t_values(i))); in __anon7c94107b1502()
1415 dim = c->MakeDim(c->Rank(c->input(i))); in ShapeShapeFn()
Dnn_ops.cc53 output_dims.push_back(c->MakeDim(val)); in FractionalPoolShapeFn()
363 new_height = c->MakeDim(size->flat<int32>()(0)); in CommonFusedConvCalculations()
364 new_width = c->MakeDim(size->flat<int32>()(1)); in CommonFusedConvCalculations()
1213 k_dim = c->MakeDim(k); in TopKShapeFn()
Dsparse_ops.cc219 DimensionHandle output_row_count = c->MakeDim(0ll); in __anona7ffb82a0a02()
Ddata_flow_ops.cc134 all_indices_constant ? c->MakeDim(max_index + 1) : c->UnknownDim()); in DynamicStitchShapeFunction()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc321 return MakeDim(0); in NumElements()
329 return MakeDim(size); in NumElements()
429 DimensionHandle d = MakeDim(value); in WithValue()
741 dims_actual.push_back(MakeDim(d)); in MakeShape()
895 dims.push_back(MakeDim(val)); in InternalMakeShapeFromTensor()
906 dims.push_back(MakeDim(val)); in InternalMakeShapeFromTensor()
929 dims[i] = MakeDim(partial_shape.dim_size(i)); in MakeShapeFromPartialTensorShape()
979 *out = MakeDim(val); in MakeDimForScalarInput()
1008 *out = MakeDim(val); in MakeDimForScalarInputWithNegativeIndexing()
1032 *out = MakeDim(v / divisor_value); in Divide()
[all …]
Dshape_inference_test.cc1195 TEST_F(ShapeInferenceTest, MakeDim) { in TEST_F() argument
1200 auto d0 = c.MakeDim(1); in TEST_F()
1201 auto d1 = c.MakeDim(1); in TEST_F()
1202 auto d2 = c.MakeDim(2); in TEST_F()
1375 EXPECT_TRUE(c.Add(d_unknown, c.MakeDim(0ll), &out).ok()); in TEST_F()
1377 EXPECT_TRUE(c.Add(d_6, c.MakeDim(0ll), &out).ok()); in TEST_F()
1387 EXPECT_TRUE(c.Add(d_6, c.MakeDim(2), &out).ok()); in TEST_F()
1390 c.Add(d_6, c.MakeDim(std::numeric_limits<int64>::max() - 6), &out).ok()); in TEST_F()
1426 EXPECT_TRUE(c.Subtract(d_unknown, c.MakeDim(0ll), &out).ok()); in TEST_F()
1428 EXPECT_TRUE(c.Subtract(d_6, c.MakeDim(0ll), &out).ok()); in TEST_F()
[all …]
Dcommon_shape_fns.cc164 c->Subtract(c->MakeDim(filter_size), 1, &window_size)); in GetWindowedOutputSizeFromDimsV2()
334 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat()
336 dims_actual[outer_c_index] = context->MakeDim(C); in MakeShapeFromFormat()
339 context->MakeDim(4); in MakeShapeFromFormat()
342 context->MakeDim(4); in MakeShapeFromFormat()
346 context->MakeDim(spatial[spatial_dim]); in MakeShapeFromFormat()
399 out_dims[GetTensorInnerFeatureDimIndex(rank, format)] = context->MakeDim(4); in ShapeFromDimensions()
1200 dims.emplace_back(c->MakeDim(1)); in ReductionShape()
1330 if (rank_x != rank_y) dim_one = c->MakeDim(1); in BroadcastBinaryOpOutputShapeFnHelper()
1406 dims->emplace_back(c->MakeDim(sizes_vec(i))); in SliceHelper()
[all …]
Dshape_inference.h500 inline DimensionHandle MakeDim(DimensionOrConstant d) { in MakeDim() function
501 return shape_manager_.MakeDim(d); in MakeDim()
504 inline DimensionHandle UnknownDim() { return MakeDim(kUnknownDim); } in UnknownDim()
667 inline DimensionHandle MakeDim(DimensionOrConstant d) { in MakeDim() function
Dshape_inference_testutil.cc245 dims.push_back(manager->MakeDim(InferenceContext::kUnknownDim)); in MakeShapeFromString()
256 dims.push_back(manager->MakeDim(dim_size)); in MakeShapeFromString()
Dshape_inference_testutil_test.cc142 c->set_output(0, c->MakeShape({c->Dim(c->input(0), 1), c->MakeDim(2), in TEST()
/external/tensorflow/tensorflow/contrib/image/ops/
Dsingle_image_random_dot_stereograms_ops.cc63 {y_dim, x_dim, colors > 256 ? c->MakeDim(3) : c->MakeDim(1)})); in __anonbda156eb0102()
Dimage_ops.cc55 height = c->MakeDim(vec(0)); in SetOutputToSizedImage()
56 width = c->MakeDim(vec(1)); in SetOutputToSizedImage()
/external/tensorflow/tensorflow/lite/experimental/microfrontend/ops/
Daudio_microfrontend_op.cc89 num_frames = ctx->MakeDim(0); in __anon68ee72bf0102()
100 DimensionHandle num_features = ctx->MakeDim(num_channels); in __anon68ee72bf0102()
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
Dreduce_slice_ops.cc48 c->Merge(c->Dim(c->input(1), 1), c->MakeDim(2), &dimhandle)); in ReduceSliceShapeFn()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1457 dims.push_back(size < 0 ? ic->UnknownDim() : ic->MakeDim(size)); in MaybeUpdateNodeContextOutput()
1670 dims.push_back(value < 0 ? ic->UnknownDim() : ic->MakeDim(value)); in MaybeTensorValueToShape()
1682 *tensors_as_shapes = ic->MakeShape({ic->MakeDim(value)}); in MaybeTensorValueToShape()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc507 : target_context->MakeDim(size)); in ConstantPartialShape()
Dshape_refiner_test.cc396 dims.emplace_back(c->MakeDim(shape_data->flat<int32>()(i))); in __anon7afc939f0602()
415 dims.emplace_back(c->MakeDim(shape_data->flat<int64>()(i))); in __anon7afc939f0702()
/external/tensorflow/tensorflow/c/
Dc_api.cc711 dim_vec.push_back(ic->MakeDim(dims[i])); in ShapeHandleFromDims()