Home
last modified time | relevance | path

Searched refs:flat_outer_dims (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc548 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local
549 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F()
550 EXPECT_EQ(60, flat_outer_dims.dimension(1)); in TEST_F()
551 EXPECT_EQ(flat_outer_dims(0, 0), 0.01f); in TEST_F()
552 EXPECT_EQ(flat_outer_dims(1, 59), 0.02f); in TEST_F()
555 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local
556 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F()
557 EXPECT_EQ(3, flat_outer_dims.dimension(1)); in TEST_F()
558 EXPECT_EQ(20, flat_outer_dims.dimension(2)); in TEST_F()
559 EXPECT_EQ(flat_outer_dims(0, 0, 0), 0.01f); in TEST_F()
[all …]
Dtensor.h418 typename TTypes<T, NDIMS>::Tensor flat_outer_dims();
519 typename TTypes<T, NDIMS>::ConstTensor flat_outer_dims() const;
884 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function
901 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function
/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc47 parent->flat_outer_dims<T>().chip(index, 0) = element.flat<T>(); in HandleElementToSlice()
54 auto parent_as_matrix = parent->flat_outer_dims<string>(); in HandleElementToSlice()
69 auto parent_as_matrix = parent->flat_outer_dims<Variant>(); in HandleElementToSlice()
84 element->flat<T>() = parent.flat_outer_dims<T>().chip(index, 0); in HandleSliceToElement()
90 element->flat<T>() = parent->flat_outer_dims<T>().chip(index, 0); in HandleSliceToElement()
96 auto parent_as_matrix = parent->flat_outer_dims<string>(); in HandleSliceToElement()
110 auto parent_as_matrix = parent->flat_outer_dims<Variant>(); in HandleSliceToElement()
/external/tensorflow/tensorflow/core/kernels/
Drandom_shuffle_op.cc93 const auto input_mat = input.flat_outer_dims<T>(); in Compute()
94 auto output_mat = output->flat_outer_dims<T>(); in Compute()
Dcwise_op_select.cc85 FastBoundsCheck(then->flat_outer_dims<T>().dimension(1), in ComputeBroadcasting()
111 func(ctx->eigen_device<Device>(), output->flat_outer_dims<T>(), in ComputeBroadcasting()
112 cond->vec<bool>(), then->flat_outer_dims<T>(), in ComputeBroadcasting()
113 else_->flat_outer_dims<T>()); in ComputeBroadcasting()
Dinplace_ops_functor_gpu.cu.cc47 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
110 auto Ty = y->flat_outer_dims<T>(); in DoInplaceOp()
145 auto Ty = y->flat_outer_dims<bool>(); in DoInplaceOp()
Dsegment_reduction_ops.cc91 auto input_flat = input.flat_outer_dims<T>(); in Compute()
114 auto output_flat = output->flat_outer_dims<T>(); in Compute()
284 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync()
493 auto output_flat = output->flat_outer_dims<T>(); in Compute()
658 auto input_flat = input.flat_outer_dims<T>(); in Compute()
688 output->flat_outer_dims<T>().setConstant(default_value_); in Compute()
694 auto output_flat = output->flat_outer_dims<T>(); in Compute()
1032 auto input_flat = input.flat_outer_dims<T>(); in Compute()
1069 auto output_flat = output->flat_outer_dims<T>(); in Compute()
Dinplace_ops.cc38 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
299 auto Tv = v.flat_outer_dims<T>(); in DoInplaceOp()
300 auto Ty = y->flat_outer_dims<T>(); in DoInplaceOp()
322 auto Tv = v.flat_outer_dims<string>(); in DoInplaceStringUpdateOp()
323 auto Ty = y->flat_outer_dims<string>(); in DoInplaceStringUpdateOp()
Dsparse_conditional_accumulator.h264 auto sum_flat = sum_tensor->flat_outer_dims<T>(); in AddToAccumGradFunction()
265 auto accum_flat = accum_val_->flat_outer_dims<T>(); in AddToAccumGradFunction()
266 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction()
324 auto accum_flat = accum_val_->flat_outer_dims<T>(); in DivideAccumGradByCounter()
Dtraining_ops.cc853 auto var_flat = var.flat_outer_dims<T>(); in DoCompute()
854 auto accum_grad_flat = accum_grad.flat_outer_dims<T>(); in DoCompute()
855 auto accum_update_flat = accum_update.flat_outer_dims<T>(); in DoCompute()
856 auto grad_flat = grad.flat_outer_dims<T>(); in DoCompute()
1039 auto var_flat = var.flat_outer_dims<T>(); in Compute()
1040 auto grad_flat = grad.flat_outer_dims<T>(); in Compute()
1388 auto var_flat = var.flat_outer_dims<T>(); in Compute()
1389 auto accum_flat = accum.flat_outer_dims<T>(); in Compute()
1390 auto grad_flat = grad.flat_outer_dims<T>(); in Compute()
1546 auto var_flat = var.flat_outer_dims<T>(); in Compute()
[all …]
Dragged_gather_op.cc38 params_dense_values_in.flat_outer_dims<VALUE_TYPE, 2>(); in WriteValueSlices()
39 auto values = values_out->flat_outer_dims<VALUE_TYPE, 2>(); in WriteValueSlices()
Ddynamic_stitch_op.cc187 const int slice_size = merged->flat_outer_dims<T>().dimension(1); in Compute()
251 auto merged_flat = merged->flat_outer_dims<T>(); in Compute()
Dscatter_op.cc125 auto params_flat = params.flat_outer_dims<T>(); in DoCompute()
213 auto params_flat = params.flat_outer_dims<T>(); in DoCompute()
Ddynamic_partition_op.cc132 out_flat.push_back(outputs[p]->flat_outer_dims<T>()); in Compute()
Dgather_nd_op.cc169 auto params_flat = params.flat_outer_dims<T, IXDIM + 1>(); \ in DoGatherNd()
Dqueue_base.cc45 auto parent_as_matrix = parent.flat_outer_dims<T>(); in HandleSliceToElement()
Dbarrier_ops.cc312 auto values_matrix = values.flat_outer_dims<T>(); in InsertOneLocked()
Dresource_variable_ops.cc786 auto params_flat = params->flat_outer_dims<T>(); in Compute()
Dlist_kernels.h997 auto input_t = input.flat_outer_dims<T, 2>(); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc196 auto gradients = gradients_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput()
204 auto hessians = hessians_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput()
277 const auto& gradients = gradients_t.flat_outer_dims<float>(); in AddToTensorAccumulator()
279 const auto& hessians = hessians_t.flat_outer_dims<float>(); in AddToTensorAccumulator()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc497 auto output_t = output->flat_outer_dims<type>(); \ in CopyPartialBatch()
498 auto value_t = value.flat_outer_dims<type>(); \ in CopyPartialBatch()
Dnuma_map_and_batch_dataset_op.cc840 auto output_t = output->flat_outer_dims<type>(); \ in CopyPartialBatch()
841 auto value_t = value.flat_outer_dims<type>(); \ in CopyPartialBatch()